Update app.py

app.py

@@ -2,7 +2,7 @@
 from doctest import OutputChecker
 import sys
 import argparse
-#import torch
+import torch
 import re
 import os
 import gradio as gr
@@ -19,7 +19,7 @@ import requests
 #from sklearn.metrics.pairwise import cosine_similarity
 
 #device = "cuda:0" if torch.cuda.is_available() else "cpu"
-model = gr.Interface.load('huggingface/sentence-transformers/stsb-distilbert-base') 
+model_sts = gr.Interface.load('huggingface/sentence-transformers/stsb-distilbert-base') 
 
 #SentenceTransformer('stsb-distilbert-base', device=device)
 
@@ -108,8 +108,8 @@ def Visual_re_ranker(caption, visual_context_label, visual_context_prob):
     caption = caption 
     visual_context_label= visual_context_label
     visual_context_prob = visual_context_prob
-    caption_emb = model.encode(caption, convert_to_tensor=True)
-    visual_context_label_emb = model.encode(visual_context_label, convert_to_tensor=True)
+    caption_emb = model_sts.encode(caption, convert_to_tensor=True)
+    visual_context_label_emb = model_sts.encode(visual_context_label, convert_to_tensor=True)
 
 
     sim =  cosine_scores = util.pytorch_cos_sim(caption_emb, visual_context_label_emb)