Update app.py

app.py

@@ -97,7 +97,10 @@ def suggest_title(component, description,pid):
             soft_prompt[k] = v
     
     generated_title = generate_title(soft_prompt, new_doc_text)
-    return f"{generated_title}",embedding
+    return {
+        "title": f"{generated_title}",
+        "embedding": embedding
+    }
 
 def infer(component,title="",description="",pid=None,mode="priority"):
     if mode=="title":
@@ -121,7 +124,7 @@ def infer(component,title="",description="",pid=None,mode="priority"):
 
     # Predict
     prediction = xgb_model.predict(test_df,iteration_range=(0, 130))
-    return f"Predicted Priority: {int(prediction[0])}"
+    return f"{int(prediction[0])}"
 
 # Gradio interface
 iface = gr.Interface(