Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -244,7 +244,7 @@ def getInputs(image_path, question, answer=""):
|
|
| 244 |
|
| 245 |
return start_input_ids, end_input_ids, image_features, attention_mask
|
| 246 |
|
| 247 |
-
model_location = "./
|
| 248 |
# print("Model location:", model_location)
|
| 249 |
|
| 250 |
model = MultimodalPhiModel.from_pretrained(model_location).to(device)
|
|
|
|
| 244 |
|
| 245 |
return start_input_ids, end_input_ids, image_features, attention_mask
|
| 246 |
|
| 247 |
+
model_location = "./MM_FT_C1_V2"
|
| 248 |
# print("Model location:", model_location)
|
| 249 |
|
| 250 |
model = MultimodalPhiModel.from_pretrained(model_location).to(device)
|