duyan2803 commited on
Commit
b8d2cfc
·
verified ·
1 Parent(s): 9d11d55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-vit-bert"
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: