Update app.py

app.py

@@ -1,14 +1,16 @@
 import gradio as gr
+import json
 from sentence_transformers import SentenceTransformer
 
-model_Q = SentenceTransformer('sentence-transformers/all-MiniLM-L12-v2')
+model = SentenceTransformer('all-MiniLM-L6-v2', device='cuda')
+model.max_seq_length = 256
 
 def text_to_embedding(text):
-    embedding = model_Q.encode(text)
-    return embedding
+    embedding = model.encode(text, device='cuda')
+    return json.dumps(embedding)
 
 inputs = gr.inputs.Textbox(default="Type text here.")
-outputs = gr.outputs.Textbox()
+outputs = gr.outputs.Label()
 
 app = gr.Interface(fn=text_to_embedding, inputs=inputs, outputs=outputs, title="Text to Embedding")
 app.launch()