rrg92 commited on
Commit
360e8ae
·
1 Parent(s): 6cd3295
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -20,6 +20,7 @@ def embed(text):
20
  return query_embedding.tolist();
21
 
22
 
 
23
  #@app.post("/v1/embeddings")
24
  #async def openai_embeddings(request: Request):
25
  # body = await request.json();
@@ -42,18 +43,21 @@ def embed(text):
42
  # }
43
  # }
44
 
 
 
 
45
  with gr.Blocks(fill_height=True) as demo:
46
  text = gr.Textbox();
47
  embeddings = gr.Textbox()
48
 
49
- text.submit(embed, [text], [embeddings]);
50
 
51
 
52
  print("Loading embedding model");
53
  Embedder = None #SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
54
 
55
 
56
- GradioApp = gr.mount_gradio_app(app, demo, path="/", ssr_mode=True);
57
 
58
  if __name__ == "__main__":
59
  uvicorn.run(GradioApp, port=7860, host="0.0.0.0")
 
20
  return query_embedding.tolist();
21
 
22
 
23
+
24
  #@app.post("/v1/embeddings")
25
  #async def openai_embeddings(request: Request):
26
  # body = await request.json();
 
43
  # }
44
  # }
45
 
46
+ def fn(text):
47
+ embed(text);
48
+
49
  with gr.Blocks(fill_height=True) as demo:
50
  text = gr.Textbox();
51
  embeddings = gr.Textbox()
52
 
53
+ text.submit(fn, [text], [embeddings]);
54
 
55
 
56
  print("Loading embedding model");
57
  Embedder = None #SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
58
 
59
 
60
+ GradioApp = gr.mount_gradio_app(app, demo, path="/", ssr_mode=False);
61
 
62
  if __name__ == "__main__":
63
  uvicorn.run(GradioApp, port=7860, host="0.0.0.0")