Update app.py

app.py

@@ -3,8 +3,8 @@ from transformers import pipeline
 
 classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7")
 
-def classify(text, *labels):
-    labels = list(labels)
+def classify(text, labels):
+    labels = labels.split(',')
     if not text or not labels:
         return []
     result = classifier(text, candidate_labels=labels)
@@ -17,10 +17,7 @@ with gr.Blocks() as demo:
     labels_input = gr.Textbox(label="Classes (comma separated)")
     
     output = gr.Dataframe(headers=["Class", "Probability"], label="Classification Results")
-
-    def update_classes(classes):
-        return classes.split(',')
-
+    
     button = gr.Button("Classify")
     
     button.click(classify, inputs=[text_input, labels_input], outputs=output)