thechaiexperiment commited on
Commit
a886035
·
1 Parent(s): 0de2895

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -225,6 +225,7 @@ def translate_text(text, source_to_target='ar_to_en'):
225
  return text
226
 
227
  def embed_query_text(query_text):
 
228
  query_embedding = embedding.encode([query_text])
229
  return query_embedding
230
 
@@ -630,3 +631,4 @@ if not init_success:
630
  if __name__ == "__main__":
631
  import uvicorn
632
  uvicorn.run(app, host="0.0.0.0", port=7860)
 
 
225
  return text
226
 
227
  def embed_query_text(query_text):
228
+ embedding = models['embedding']
229
  query_embedding = embedding.encode([query_text])
230
  return query_embedding
231
 
 
631
  if __name__ == "__main__":
632
  import uvicorn
633
  uvicorn.run(app, host="0.0.0.0", port=7860)
634
+