File size: 498 Bytes
47ad9af
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import gradio as gr
from transformers import pipeline

model = pipeline("text-generation", model="gpt2")  # مدل GPT-2

def generate_text(message):
    response = model(message, max_length=50, num_return_sequences=1)
    return response[0]['generated_text']

iface = gr.Interface(
    fn=generate_text,
    inputs=gr.Textbox(label="پیام ورودی"),
    outputs=gr.Textbox(label="پاسخ مدل"),
    title="چت با GPT-2"
)
iface.launch(share=True)  # share=True برای URL عمومی