Experimenting with few-shot classification.

app.py

@@ -62,8 +62,10 @@ 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)
-
-    return zero_shot_classification
+    if 'task' in data and data['task'] == 'few_shot_classification':
+        return few_shot_classification(data)
+    else:
+        return zero_shot_classification(data)
 
 def zero_shot_classification(data):
     results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])