Update app.py

app.py

@@ -48,6 +48,7 @@ def classify_text(
             tokenizer = AutoTokenizer.from_pretrained(model_name)
             classifier = pipeline("zero-shot-classification",
                                   model=model_name,
+                                  device=device,
                                   tokenizer=tokenizer,
                                   trust_remote_code=True
                                   )