Update app.py

app.py

@@ -14,8 +14,8 @@ def get_format_response(image,question,selected_model):
     processor = processor.from_pretrained(model_name)
     model = model_class.from_pretrained(model_name)
     encoding = processor(image, question, return_tensors="pt")
-    outputs = model(**encoding)
     if selected_model=='ViLT':
+        outputs = model(**encoding)
         logits = outputs.logits
         idx = logits.argmax(-1).item()
         answer = model.config.id2label[idx]