Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,7 @@ class VQAModel(nn.Module):
|
|
26 |
return output
|
27 |
|
28 |
# Load mô hình từ Hugging Face Hub
|
29 |
-
repo_id = "duyan2803/vqa-model-
|
30 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
31 |
|
32 |
try:
|
|
|
26 |
return output
|
27 |
|
28 |
# Load mô hình từ Hugging Face Hub
|
29 |
+
repo_id = "duyan2803/vqa-model-vilt-bert-color-optim"
|
30 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
31 |
|
32 |
try:
|