Rename app_main.py to app.py

app_main.py → app.py

@@ -10,11 +10,11 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
 # Initialize the model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("ManishThota/SparrowVQE",
+model = AutoModelForCausalLM.from_pretrained("rrymn/SparrowVQE",
                                              torch_dtype=torch.float16, 
                                              device_map="auto",
                                              trust_remote_code=True)
-tokenizer = AutoTokenizer.from_pretrained("ManishThota/SparrowVQE", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("rrymn/SparrowVQE", trust_remote_code=True)
 
 def predict_answer(image, question, max_tokens=100):
     #Set inputs