Fixing model loading

app.py

@@ -93,11 +93,12 @@ def predict(img):
     models_list = ['clip-sd.pth', 'clip-glide.pth', 'clip-ld.pth']
     modality = "Image+Text"
     for model_path in models_list:
-        model = get_model('clip-sd.pth', modality)
+        model = get_model(model_path, modality)
         tensor = preprocessing(img, 224)
         input_tensor = tensor.view(1, 3, 224, 224)
         with torch.no_grad():
-            out = model(input_tensor)
+            out = model(input_tensor, caption)
+            print('------------>', out)
             prediction.append(out)
 
     if prediction[0] > 0.5: