Spaces:

merve
/

chatgpt-prompt-generator-v12

Runtime error

merve HF Staff commited on Jan 12, 2024

Commit

b7cfa6a

verified ·

1 Parent(s): 1b3b6ca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import torch
 tokenizer = AutoTokenizer.from_pretrained("merve/chatgpt-prompt-generator-v12")
 model = AutoModelForSeq2SeqLM.from_pretrained("merve/chatgpt-prompt-generator-v12", from_tf=True)
-device = "cuda:0" if torch.cuda.is_available() else "cpu"
 model.to(device)
 @spaces.GPU

 tokenizer = AutoTokenizer.from_pretrained("merve/chatgpt-prompt-generator-v12")
 model = AutoModelForSeq2SeqLM.from_pretrained("merve/chatgpt-prompt-generator-v12", from_tf=True)
+device = "cuda" if torch.cuda.is_available() else "cpu"
 model.to(device)
 @spaces.GPU