Add application file

app.py

@@ -0,0 +1,17 @@
+from transformers import pipeline, Conversation
+import gradio as gr
+
+chatbot = pipeline(model="Falconsai/medical_summarization")
+
+message_list = []
+response_list = []
+
+def vanilla_chatbot(message, history):
+    conversation = Conversation(text=message, past_user_inputs=message_list, generated_responses=response_list)
+    conversation = chatbot(conversation)
+
+    return conversation.generated_responses[-1]
+
+demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Test medical summarization chatbot", description="Enter text to summarize.")
+
+demo_chatbot.launch()

requirements.txt

@@ -0,0 +1,3 @@
+gradio==3.39.0
+transformers==4.31.0
+torch==2.0.1