Messing with fastAPI.

app.py

@@ -52,9 +52,9 @@ def classify(data_string, request: gradio.Request):
     # else:
     start_time = time.time()
     result = {}
-    if (data['cpu'])
+    if (data['cpu']):
         result = zero_shot_classification_cpu(data)
-    else
+    else:
         result = zero_shot_classification_gpu(data)
     print(f"Classification @ [{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] took {time.time() - start_time}.")
     return json.dumps(result)