fix

app.py

@@ -25,7 +25,7 @@ def load_model():
         num_labels=len(tag_to_index),
         id2label=index_to_tag,
         label2id=tag_to_index
-    ).to(torch.device('cuda'))
+    )
 
 model = load_model()