File size: 372 Bytes
e13365d
 
 
314ea65
e13365d
 
 
a005d28
e13365d
 
401e3df
1
2
3
4
5
6
7
8
9
10
11
12
import gradio as gr
from transformers import pipeline

generator = pipeline("summarization", model="Falconsai/medical_summarization")

def generate_response(input_text):
    response = generator(input_text, max_length=50)
    return response[0]['generated_text1']

interface = gr.Interface(fn=generate_response, inputs="text", outputs="text")
interface.launch(share=True)