summay

app.py

@@ -1,7 +1,7 @@
 from transformers import pipeline, Conversation
 import gradio as gr
 
-chatbot = pipeline(model="Falconsai/medical_summarization")
+chatbot = pipeline("summarization", model="Falconsai/medical_summarization")
 
 message_list = []
 response_list = []
@@ -12,7 +12,10 @@ def vanilla_chatbot(message, history):
 
     return conversation.generated_responses[-1]
 
-demo_chatbot = gr.ChatInterface(vanilla_chatbot, title="Test medical summarization chatbot", description="Enter text to summarize.")
+def summarizer_bot(message, history):
+    return chatbot(message, min_length=5, max_length=500)[0]['summary_text']
+
+demo_chatbot = gr.ChatInterface(summarizer_bot, title="Test medical summarization chatbot", description="Enter text to summarize.")
 
 # def responseFun(message, history):
 #     return history[len(history) - 1]