Update app.py

app.py

@@ -15,11 +15,11 @@ from io import StringIO
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 # Initialize the model and tokenizer
-model = AutoModelForCausalLM.from_pretrained("ManishThota/SparrowVQE",
+model = AutoModelForCausalLM.from_pretrained("ManishThota/CustomModel",
                                              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("ManishThota/CustomModel", trust_remote_code=True)