Update app.py

app.py

@@ -70,7 +70,7 @@ def generate_answer_vilt(processor, model, image, question):
     logitsList = outputs.logits.flatten().tolist()
     print(f"predicted_class_idx_in_list = {logitsList.index(max(logitsList))}")
     m = max(logitsList)
-    s = -math.infinity 
+    s = -math.inf 
     for logit in logitsList:
         if s <= logit < m:
             s = logit