Update app.py

app.py

@@ -134,7 +134,7 @@ prompt = st.text_input(" ")
 
 
 if prompt != "":
-    sbert_saved_model = torch.load("Sbert_saved_model")
+    sbert_saved_model = torch.load("Sbert_saved_model",map_location=torch.device('cpu'))
     model = sbert_saved_model.to("cpu")
     tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-mpnet-base-v2")
     pipe = TextClassificationPipeline(model= model, tokenizer=tokenizer, return_all_scores=True)