Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from transformers import pipeline
 
 # Load the model
-pipe = pipeline("text2text-generation", model="google/flan-t5-base")
+pipe = pipeline("text2text-generation", model="declare-lab/flan-alpaca-base")
 
 def convert_to_json(text):
-    prompt = f"Convert the following sentence into a JSON object with keys: userId, type, category, amount, date, note. ": {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