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)
|