Spaces:

OzoneAsai
/

GPTsan2

Sleeping

OzoneAsai commited on Jun 1, 2023

Commit

7b70bdf

1 Parent(s): b5d7b7a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,6 +10,7 @@ result = os.system("pip install transformers")
 from transformers import AutoModel, AutoTokenizer, trainer_utils
 import gradio as gr
 device = "cpu"
 model = AutoModel.from_pretrained("Tanrei/GPTSAN-japanese").to(device)

 from transformers import AutoModel, AutoTokenizer, trainer_utils
 import gradio as gr
+import psutil
 device = "cpu"
 model = AutoModel.from_pretrained("Tanrei/GPTSAN-japanese").to(device)