Update app.py

app.py

@@ -46,13 +46,7 @@ def classify_text(
 
         if model_name in CUSTOM_MODELS:
             tokenizer = AutoTokenizer.from_pretrained(model_name)
-            classifier = pipeline("zero-shot-classification",
-                                  model=model_name,
-                                  device=device,
-                                  tokenizer=tokenizer,
-                                  trust_remote_code=True
-                                  )
-
+            classifier = pipeline("zero-shot-classification", model=model_name, device=device, tokenizer=tokenizer, trust_remote_code=True)
         else:
             classifier = pipeline("zero-shot-classification", model=model_name, device=device)