Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import pipeline
 pipe = pipeline("text2text-generation", model="google/flan-t5-base")
 
 def convert_to_json(text):
-    prompt = f"Convert this to JSON: {text}"
+    prompt = f"Convert the following sentence into a JSON object with keys: userId, type, category, amount, date, note. ": {text}"
     output = pipe(prompt, max_new_tokens=200)[0]["generated_text"]
     return output