ManishThota commited on
Commit
f8dcf83
1 Parent(s): c52e238

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,11 +15,11 @@ torch.set_default_tensor_type('torch.cuda.FloatTensor')
15
 
16
 
17
  # Initialize the model and tokenizer
18
- model = AutoModelForCausalLM.from_pretrained("ManishThota/Sparrow",
19
  torch_dtype=torch.float16,
20
  device_map="auto",
21
  trust_remote_code=True)
22
- tokenizer = AutoTokenizer.from_pretrained("ManishThota/Sparrow", trust_remote_code=True)
23
 
24
  def predict_answer(image, question, max_tokens):
25
  #Set inputs
 
15
 
16
 
17
  # Initialize the model and tokenizer
18
+ model = AutoModelForCausalLM.from_pretrained("ManishThota/SparrowVQE",
19
  torch_dtype=torch.float16,
20
  device_map="auto",
21
  trust_remote_code=True)
22
+ tokenizer = AutoTokenizer.from_pretrained("ManishThota/SparrowVQE", trust_remote_code=True)
23
 
24
  def predict_answer(image, question, max_tokens):
25
  #Set inputs