CPU fallback.

app.py

@@ -55,11 +55,12 @@ def classify(data_string, request: gradio.Request):
     try:
         if 'cpu' not in data:
             result = zero_shot_classification_gpu(data)
+            print(f"GPU Classification @ [{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] took {time.time() - start_time}.")
     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}.")
+        result = zero_shot_classification_cpu(data)
+        print(f"CPU Classification @ [{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] took {time.time() - start_time}.")
     return json.dumps(result)
 
 def zero_shot_classification_cpu(data):