EliHPLaptop
Create app.py file
430d0f0
raw
history blame contribute delete
594 Bytes
from transformers import pipeline
model = pipeline("summarization", model="eliori/dialogue_summarization-finetuned")
def summarize(prompt):
completion = model(prompt)[0]["summary_text"]
return completion
title = "Write or Paste in a Dialogue"
description = """
This model is used to summarize dialogues.
"""
import gradio as gr
# from huggingface_hub import InferenceClient
gr.Interface(
fn=summarize,
inputs="textbox",
outputs="text",
title=title,
description=description,
examples=[["#Person1#: (dialogue) #Person2#: (response to dialogue)"]],
).launch()