File size: 865 Bytes
29e745d
c4392b4
29e745d
c4392b4
 
29e745d
c4392b4
 
29e745d
c4392b4
 
 
29e745d
c4392b4
 
 
 
29e745d
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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()