Hugging Face's logo

Spaces:
OzoneAsai
/
GPTsan2
Sleeping

App Files Community
OzoneAsai commited on
Commit
e6a63bf
·
1 Parent(s): 47e3ba6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ result = os.system("pip install transformers")
11
  from transformers import AutoModel, AutoTokenizer, trainer_utils
12
  import gradio as gr
13
 
14
- device = "cuda"
15
  model = AutoModel.from_pretrained("Tanrei/GPTSAN-japanese").to(device)
16
  tokenizer = AutoTokenizer.from_pretrained("Tanrei/GPTSAN-japanese")
17
  trainer_utils.set_seed(30)
 
11
  from transformers import AutoModel, AutoTokenizer, trainer_utils
12
  import gradio as gr
13
 
14
+ device = "cpu"
15
  model = AutoModel.from_pretrained("Tanrei/GPTSAN-japanese").to(device)
16
  tokenizer = AutoTokenizer.from_pretrained("Tanrei/GPTSAN-japanese")
17
  trainer_utils.set_seed(30)