Update app.py

app.py

@@ -2,26 +2,30 @@ import gradio as gr
 from sentence_transformers import SentenceTransformer
 import torch
 
-# Load model with trust_remote_code=True for perplexity models
 model = SentenceTransformer(
     'perplexity-ai/pplx-embed-v1-4b',
     trust_remote_code=True,
-    device='cpu'  # Force CPU for free tier
+    device='cpu'
 )
 
-def embed_text(texts):
-    if isinstance(texts, str):
-        texts = [texts]
-    embeddings = model.encode(texts, normalize_embeddings=True)
-    return embeddings.tolist()
+def embed_text(text):
+    if not text or text.strip() == "":
+        return "Please enter single text input"
+    
+    embedding = model.encode([text], normalize_embeddings=True)[0]
+    return embedding.tolist()
 
 demo = gr.Interface(
     fn=embed_text,
-    inputs=gr.Textbox(label="Enter text(s), one per line", lines=3, placeholder="Your text here..."),
-    outputs=gr.Textbox(label="Embeddings (JSON)", lines=10),
-    title="pplx-embed-v1-4b API ✅",
-    description="Free CPU embedding inference"
+    inputs=gr.Textbox(
+        label="Single Text Input", 
+        lines=1,
+        placeholder="enter here..."
+    ),
+    outputs=gr.Textbox(label="Embedding (JSON)", lines=10),
+    title="text-to-embedding",
+    description="simple text-to-embedding, using pplx-embed-v1-4b"
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=False, show_error=True)