Update app.py

app.py

@@ -9,7 +9,7 @@ from transformers import MllamaForConditionalGeneration, AutoProcessor
 import torch
 from accelerate import init_empty_weights
 # Load environment variables
-
+from transformers import AutoProcessor, AutoModelForPreTraining
 # Configure Gemini API
 # genai.configure(api_key=os.getenv("gkey2"))
 
@@ -27,12 +27,14 @@ prompt="<|image|><|begin_of_text|>You are a helpful assistant. Please respond to
 # model = Ollama(model="llama3.2")
 model_id = "meta-llama/Llama-3.2-11B-Vision"
 
-model = MllamaForConditionalGeneration.from_pretrained(
-    model_id,
-    torch_dtype=torch.bfloat16,
-    device_map="auto",
-)
-processor = AutoProcessor.from_pretrained(model_id)
+# model = MllamaForConditionalGeneration.from_pretrained(
+#     model_id,
+#     torch_dtype=torch.bfloat16,
+#     device_map="auto",
+# )
+# processor = AutoProcessor.from_pretrained(model_id)
+processor = AutoProcessor.from_pretrained("meta-llama/Llama-3.2-11B-Vision")
+model = AutoModelForPreTraining.from_pretrained("meta-llama/Llama-3.2-11B-Vision")
 
 # Define function to get response from the model
 def get_gemin_response(input_text, img):