Katake commited on
Commit
45c54c3
·
verified ·
1 Parent(s): 7a2222a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +28 -0
app.py CHANGED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM
3
+
4
+ model_name = "burhanmt/destilgpt2-turkish" # Türkçe DistilGPT2 modeli (hafif ve hızlı)
5
+
6
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
7
+ model = AutoModelForCausalLM.from_pretrained(model_name)
8
+
9
+ def generate_text(prompt):
10
+ inputs = tokenizer(prompt, return_tensors="pt")
11
+ outputs = model.generate(
12
+ **inputs,
13
+ max_length=120,
14
+ do_sample=True,
15
+ top_p=0.9,
16
+ temperature=0.8
17
+ )
18
+ return tokenizer.decode(outputs[0], skip_special_tokens=True)
19
+
20
+ demo = gr.Interface(
21
+ fn=generate_text,
22
+ inputs=gr.Textbox(label="Türkçe metin başlangıcı"),
23
+ outputs=gr.Textbox(label="Üretilen metin"),
24
+ title="Türkçe Metin Üretici (DistilGPT2)",
25
+ description="Hafif ve hızlı Türkçe metin üretme modeli."
26
+ )
27
+
28
+ demo.launch()