Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoProcessor, AutoModelForCausalLM, BlipForQuestionAnswering, ViltForQuestionAnswering
 import torch
+import math
 
 torch.hub.download_url_to_file('http://images.cocodataset.org/val2017/000000039769.jpg', 'cats.jpg')
 torch.hub.download_url_to_file('https://huggingface.co/datasets/nielsr/textcaps-sample/resolve/main/stop_sign.png', 'stop_sign.png')
@@ -68,7 +69,7 @@ def generate_answer_vilt(processor, model, image, question):
     predicted_class_idx = outputs.logits.argmax(-1).item()
     logitsList = outputs.logits.tolist()
     print(logitsList)
-    maybeProbsList = [exp(i) for i in logitsList]
+    maybeProbsList = [math.exp(i) for i in logitsList]
     return model.config.id2label[predicted_class_idx]