Hugging Face's logo

Spaces:
tamatwi
/
shortgpu_Qwen2-7B-Instruct
Runtime error

App Files Community
tamatwi commited on
Commit
32bfed6
·
verified ·
1 Parent(s): 1403c5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import gradio as gr
2
- from transformers import pipeline
3
  import spaces
4
  @spaces.GPU
5
  # generator = pipeline('text-generation', model='distilgpt2', device=-1)
6
- model_name = "distilgpt2"
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=-1)
9
 
 
1
  import gradio as gr
2
+ from transformers import pipeline, AutoTokenizer
3
  import spaces
4
  @spaces.GPU
5
  # generator = pipeline('text-generation', model='distilgpt2', device=-1)
6
+ model_name = "rinna/japanese-gpt2-medium"
7
  tokenizer = AutoTokenizer.from_pretrained(model_name)
8
  generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=-1)
9