Experimenting with few-shot classification.

app.py

@@ -62,7 +62,7 @@ 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 data['task'] == 'few_shot_classification':
+    if 'task' in data and data['task'] == 'few_shot_classification':
         return few_shot_classification
     else:
         return zero_shot_classification