Messing with fastAPI.

app.py

@@ -36,7 +36,6 @@ classifier = pipeline(task="zero-shot-classification", model=model_name, tokeniz
 # classifier = pipeline(task="zero-shot-classification", model=model_name, tokenizer=tokenizer_name)
 
 def classify(data_string, request: gradio.Request):
-    print(f"classify")
     if request:
         if request.headers["origin"] not in ["https://statosphere-3704059fdd7e.c5v4v4jx6pq5.win", "https://crunchatize-77a78ffcc6a6.c5v4v4jx6pq5.win", "https://crunchatize-2-2b4f5b1479a6.c5v4v4jx6pq5.win", "https://tamabotchi-2dba63df3bf1.c5v4v4jx6pq5.win", "https://ravenok-statosphere-backend.hf.space", "https://lord-raven.github.io"]:
             return "{}"
@@ -50,9 +49,7 @@ def classify(data_string, request: gradio.Request):
     # else:
     start_time = time.time()
     result = zero_shot_classification(data)
-    end_time = time.time()
-    elapsed_time = end_time - start_time
-    print(f"classification took {elapsed_time}.")
+    print(f"classification took {time.time() - start_time}.")
     return json.dumps(result)
 
 @spaces.GPU(duration=3)
@@ -84,4 +81,6 @@ gradio_interface = gradio.Interface(
     outputs = gradio.Textbox()
 )
 
+app.mount("/gradio", gradio_interface)
+
 gradio_interface.launch()