Update app.py

app.py

@@ -3,8 +3,11 @@ from transformers import pipeline
 
 # Load the large language model (LLM)
 try:
-    print("Loading the language model...")
-    llm_pipeline = pipeline("text-generation", model = AutoModelForPreTraining.from_pretrained("meta-llama/Llama-3.2-11B-Vision-Instruct")  # You can use a different model here
+# Load model directly
+from transformers import AutoProcessor, AutoModelForPreTraining
+
+processor = AutoProcessor.from_pretrained("meta-llama/Llama-3.2-11B-Vision-Instruct")
+model = AutoModelForPreTraining.from_pretrained("meta-llama/Llama-3.2-11B-Vision-Instruct") # You can use a different model here
     print("Model loaded successfully!")
 except Exception as e:
     print(f"Error loading model: {e}")