fix
Browse files
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(
|
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=
|
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")
|