fixed bugs of prediction

app.py

@@ -89,6 +89,7 @@ def get_caption(image):
 
 def predict(img):
     caption = get_caption(img)
+    print("caption")
     prediction=[]
     models_list = ['clip-sd.pth', 'clip-glide.pth', 'clip-ld.pth']
     modality = "Image+Text"
@@ -99,7 +100,7 @@ def predict(img):
         with torch.no_grad():
             out = model(input_tensor, caption)
             print('------------>', out)
-            prediction.append(out)
+            prediction.append(out.item())
 
     if prediction[0] > 0.5:
         return "Fake"