Update app.py

app.py

@@ -7,7 +7,7 @@ def load_qa_model():
     return pipeline("question-answering", model="bert-large-uncased-whole-word-masking-finetuned-squad")
 
 def load_classifier_model():
-    return pipeline("Sentiment Analysis", model="MoritzLaurer/deberta-v3-base-zeroshot-v1.1-all-33")
+    return pipeline("zero-shot-classification", model="MoritzLaurer/deberta-v3-base-zeroshot-v1.1-all-33")
 
 def load_translator_model(target_language):
     try:
@@ -91,7 +91,7 @@ with gr.Blocks() as demo:
     with gr.Tab("Single-Models"):
         gr.Markdown("This tab is for single models demonstration.")
 
-        task_select_single = gr.Dropdown(["Question Answering", "Sentiment Analysis", "Translation", "Text Generation", "Summarization"], label="Select Task")
+        task_select_single = gr.Dropdown(["Question Answering", "Zero-Shot Classification", "Translation", "Text Generation", "Summarization"], label="Select Task")
         input_text_single = gr.Textbox(label="Input Text")
 
         # Additional inputs for specific tasks
@@ -109,7 +109,7 @@ with gr.Blocks() as demo:
                     label_input_single: gr.update(visible=False),
                     target_language_input_single: gr.update(visible=False)
                 }
-            elif task == "Sentiment Analysis":
+            elif task == "Zero-Shot Classification":
                 return {
                     context_input_single: gr.update(visible=False),
                     label_input_single: gr.update(visible=True),
@@ -146,7 +146,7 @@ with gr.Blocks() as demo:
         def execute_task_single(task, input_text, context, labels, target_language):
             if task == "Question Answering":
                 return process_qa(context=context, question=input_text)
-            elif task == "Sentiment Analysis":
+            elif task == "Zero-Shot Classification":
                 if not labels:
                     return "Please provide labels for classification."
                 return process_classifier(text=input_text, labels=labels)