Update app.py

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 import torch
 import math
 import httpcore
+import pickle
 setattr(httpcore, 'SyncHTTPTransport', 'AsyncHTTPProxy')
 
 """
@@ -30,9 +31,11 @@ def respond(
     messages.append({"role": "assistant", "content": "I'd be happy to help! Please go ahead and provide the sentence you'd like me to analyze. Please specify whether you're referencing a particular verse or hadith (Prophetic tradition) from the Quran or Hadith, or if you're asking me to analyze a general statement."})
 
     #adding fatwa references
-    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-    selected_references = torch.load('selected_references.sav', map_location=torch.device(device))
-    encoded_questions = torch.load('encoded_questions.sav', map_location=torch.device(device))
+    #device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    #selected_references = torch.load('selected_references.sav', map_location=torch.device(device))
+    #encoded_questions = torch.load('encoded_questions.sav', map_location=torch.device(device))
+    selected_references = pickle.load(open('selected_references.sav','rb'))
+    encoded_questions = pickle.load(open('encoded_questions.sav','rb'))
     
     task = 'Given a web search query, retrieve relevant passages that answer the query'
     queries = [
@@ -58,7 +61,9 @@ def respond(
             try:
                 print(index)
                 print(f'{row["user"]}')
-                user = translator.translate(f'{row["user"]}', src='ar', dest='en').text
+                translated = translator.translate(f'{row["user"]}', src='ar', dest='en')
+                print(translated)
+                user = translated.text
                 print(user)
                 #print(row['assistant'])
                 assistant = translator.translate(row['assistant']).text