Update app.py

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 from transformers import AutoTokenizer
 
 def greet(prompt):
-    tokenizer = AutoTokenizer.from_pretrained('bert-base-cased', return_tensors='tf')
+    pipe = pipeline("conversational", model="Amod/falcon7b-fine-tuned-therapy-merged", trust_remote_code=True)
     
-    return tokenizer(prompt)
+    return pipe(prompt)
 
 iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 iface.launch()