Spaces:

tamatwi
/

shortrinnashort

App Files Files Community

tamatwi commited on Jun 25

Commit

32bfed6

•

1 Parent(s): 1403c5e

Update app.py

Files changed (1) hide show

app.py +2 -2

app.py CHANGED Viewed

@@ -1,9 +1,9 @@
 import gradio as gr
-from transformers import pipeline
 import spaces
 @spaces.GPU
 # generator = pipeline('text-generation', model='distilgpt2', device=-1)
-model_name = "distilgpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=-1)

 import gradio as gr
+from transformers import pipeline, AutoTokenizer
 import spaces
 @spaces.GPU
 # generator = pipeline('text-generation', model='distilgpt2', device=-1)
+model_name = "rinna/japanese-gpt2-medium"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 generator = pipeline('text-generation', model=model_name, tokenizer=tokenizer, device=-1)