Spaces:

Thouph
/

prompt2tag-qwen2-0.5b-v0.1-demo

Sleeping

Thouph commited on Jun 24

Commit

d93389c

•

1 Parent(s): 8e922f8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 from datetime import datetime
 torch.set_grad_enabled(False)
-model = Qwen2ForSequenceClassification.from_pretrained("Thouph/prompt2tag-qwen2-0.5b-v0.1", num_labels = 9940, map_location=torch.device('cpu'))
 model.eval()
 tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B")

 from datetime import datetime
 torch.set_grad_enabled(False)
+model = Qwen2ForSequenceClassification.from_pretrained("Thouph/prompt2tag-qwen2-0.5b-v0.1", num_labels = 9940, device_map="cpu")
 model.eval()
 tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2-0.5B")