mohammadshahabiy commited on
Commit
2a79d5f
·
verified ·
1 Parent(s): 46cf2c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
 
4
- # بارگذاری مدل ParsGPT
5
- model_name = "path/to/parsgpt" # آدرس مدل ParsGPT در Hugging Face یا محلی
6
 
7
  # بارگذاری توکن‌ایزر و مدل
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
3
 
4
+ # بارگذاری مدل GPT-2-Persian
5
+ model_name = "bolbolzaban/gpt2-persian"
6
 
7
  # بارگذاری توکن‌ایزر و مدل
8
  tokenizer = AutoTokenizer.from_pretrained(model_name)