Upload app.py

app.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 from doctest import OutputChecker
 import sys
-#import torch
+import torch
 import re
 import os
 import gradio as gr
@@ -17,7 +17,7 @@ from sentence_transformers import SentenceTransformer, util
 #from sentence_transformers import SentenceTransformer, util
 #from sklearn.metrics.pairwise import cosine_similarity
 
-#device = "cuda:0" if torch.cuda.is_available() else "cpu"
+
 #model_sts = gr.Interface.load('huggingface/sentence-transformers/stsb-distilbert-base') 
 
 model_sts = SentenceTransformer('stsb-distilbert-base')