Experimenting with few-shot classification.

app.py

@@ -62,10 +62,8 @@ def classify(data_string, request: gradio.Request):
         if request.headers["origin"] not in ["https://statosphere-3704059fdd7e.c5v4v4jx6pq5.win", "https://crunchatize-77a78ffcc6a6.c5v4v4jx6pq5.win", "https://ravenok-statosphere-backend.hf.space"]:
             return "{}"
     data = json.loads(data_string)
-    if 'task' in data and data['task'] == 'few_shot_classification':
-        return few_shot_classification
-    else:
-        return zero_shot_classification
+
+    return zero_shot_classification
 
 def zero_shot_classification(data):
     results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])