rrg92 commited on
Commit
96338c0
·
1 Parent(s): b45b188
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -15,9 +15,9 @@ app = FastAPI()
15
 
16
  @spaces.GPU
17
  def embed(text):
18
-
19
- query_embedding = Embedder.encode(text)
20
- return query_embedding.tolist();
21
 
22
 
23
 
@@ -54,7 +54,7 @@ with gr.Blocks(fill_height=True) as demo:
54
 
55
 
56
  print("Loading embedding model");
57
- Embedder = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
58
 
59
  # demo.run_startup_events()
60
 
@@ -68,7 +68,7 @@ Embedder = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
68
  #)
69
 
70
  print("Demo run...");
71
- (app,url,other) = demo.launch(prevent_thread_lock=True);
72
 
73
  print("Mounting app...");
74
  GradioApp = gr.mount_gradio_app(app, demo, path="/", ssr_mode=False);
 
15
 
16
  @spaces.GPU
17
  def embed(text):
18
+ return [0,1]
19
+ #query_embedding = Embedder.encode(text)
20
+ #return query_embedding.tolist();
21
 
22
 
23
 
 
54
 
55
 
56
  print("Loading embedding model");
57
+ Embedder = None #SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
58
 
59
  # demo.run_startup_events()
60
 
 
68
  #)
69
 
70
  print("Demo run...");
71
+ (app2,url,other) = demo.launch(prevent_thread_lock=True);
72
 
73
  print("Mounting app...");
74
  GradioApp = gr.mount_gradio_app(app, demo, path="/", ssr_mode=False);