CPU fallback.

app.py

@@ -52,9 +52,12 @@ def classify(data_string, request: gradio.Request):
     # else:
     start_time = time.time()
     result = {}
-    if ('cpu' in data):
-        result = zero_shot_classification_cpu(data)
-    else:
+    try:
+        if 'cpu' not in data:
+            result = zero_shot_classification_gpu(data)
+    except Exception as e:
+        print(f"GPU classification failed: {e}\nFall back to CPU.")
+    if not result:
         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)