Update app.py

app.py

@@ -2,12 +2,13 @@ import gradio as gr
 from transformers import pipeline
 import concurrent.futures
 import ktrain
+import time
 
 zero_shot = pipeline("zero-shot-classification")
 distilbert = ktrain.load_predictor("models/distilbert-base-uncased-finetuned-internet-provider")
 
 def zero_shot_predict(text):
-    labels = ["Slow Connectivity", "Billing", "Setup", "No Connectivity"]
+    labels = ["Slow Connection", "Billing", "Setup", "No Connectivity"]
     preds = zero_shot(text, candidate_labels=labels)
     return {label: float(pred) for label, pred in zip(preds["labels"], preds["scores"])}
 
@@ -17,6 +18,7 @@ def distilbert_predict(text):
     return {label: float(pred) for label, pred in zip(labels, preds)}
 
 def predict(text):
+    time.sleep(1)
     with concurrent.futures.ThreadPoolExecutor(max_workers=2) as executor:
         zero_shot_future = executor.submit(zero_shot_predict, text)
         distilbert_future = executor.submit(distilbert_predict, text)