text2image-Pro / app.py
Rooni's picture
Update app.py
c4392b4
raw
history blame
865 Bytes
import gradio as gr
from transformers import pipeline
# Выберите модель HuggingFace, которую вы хотите использовать
model_name = "gpt-2"
# Создайте "pipeline" для вашей модели. В этом примере мы используем текстовый генератор.
model = pipeline('text-generation', model=model_name)
# Определите функцию, которая будет использоваться Gradio для интерактивного ввода и вывода
def generate_text(inp):
return model(inp)[0]['generated_text']
# Создайте интерфейс Gradio. Вы можете настроить его под свои нужды.
iface = gr.Interface(fn=generate_text, inputs="text", outputs="text")
# Запустите интерфейс
iface.launch()