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
|