Spaces:
Paused
Paused
Update app.py
Browse files
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/
|
19 |
torch_dtype=torch.float16,
|
20 |
device_map="auto",
|
21 |
trust_remote_code=True)
|
22 |
-
tokenizer = AutoTokenizer.from_pretrained("ManishThota/
|
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
|