Update app.py

app.py

@@ -3,19 +3,19 @@
 # from firebase_admin import firestore
 import io
 from fastapi import FastAPI, File, UploadFile
-from werkzeug.utils import secure_filename
+# from werkzeug.utils import secure_filename
 # import speech_recognition as sr
 import subprocess
 import os
 import requests
 import random
 import pandas as pd
-from pydub import AudioSegment
+# from pydub import AudioSegment
 from datetime import datetime
 from datetime import date
-import numpy as np
+# import numpy as np
 # from sklearn.ensemble import RandomForestRegressor
-import shutil
+# import shutil
 import json
 # from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
 from pydantic import BaseModel
@@ -28,7 +28,7 @@ from fastapi import Form
 # from transformers import AutoModelForSequenceClassification
 # from transformers import TFAutoModelForSequenceClassification
 # from transformers import AutoTokenizer, AutoConfig
-import numpy as np
+# import numpy as np
 # from scipy.special import softmax
 from sentence_transformers import SentenceTransformer
 
@@ -36,10 +36,11 @@ from sentence_transformers import SentenceTransformer
 
 
 # model = SentenceTransformer('flax-sentence-embeddings/all_datasets_v4_MiniLM-L6')
-model = SentenceTransformer("sentence-transformers/all-roberta-large-v1")
-
+# model = SentenceTransformer("sentence-transformers/all-roberta-large-v1")
+# model =SentenceTransformer("intfloat/multilingual-e5-large")
 
 
+model = SentenceTransformer('intfloat/multilingual-e5-large')
 
 
 class Query(BaseModel):
@@ -91,7 +92,10 @@ async def startup_event():
 async def get_answer(q: Query ):
     
     text = q.text
-    text_e = model.encode(text)
+    # text_e = model.encode(text)
+    input_texts = [text]
+    embeddings = model.encode(input_texts)
+    text_e = embeddings[0]
     
     
     dict={ }