Update app.py

app.py

@@ -7,6 +7,8 @@ from sentence_transformers import SentenceTransformer
 import chromadb
 from datasets import load_dataset
 import gradio as gr
+import torch
+
 
 
 # Load model directly