Add ZeroSpace and share=True (#8)

- Add ZeroSpace and share=True (972a79008dcc2c05ef1bfdb602a7d480e378ff4a)


Co-authored-by: Francesco Giannuzzo <[email protected]>

app.py

@@ -1,6 +1,16 @@
 import gradio as gr
 import pandas as pd
 import os
+# https://discuss.huggingface.co/t/issues-with-sadtalker-zerogpu-spaces-inquiry-about-community-grant/110625/10
+if os.environ.get("SPACES_ZERO_GPU") is not None:
+    import spaces
+else:
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            def wrapper(*args, **kwargs):
+                return func(*args, **kwargs)
+            return wrapper
 import sys
 from qatch.connectors.sqlite_connector import SqliteConnector
 from qatch.generate_dataset.orchestrator_generator import OrchestratorGenerator
@@ -13,6 +23,10 @@ import plotly.express as px
 import plotly.graph_objects as go
 import plotly.colors as pc
 
+@spaces.GPU
+def model_prediction():
+  pass
+
 with open('style.css', 'r') as file:
     css = file.read()
 
@@ -826,4 +840,4 @@ with gr.Blocks(theme='d8ahazard/rd_blue', css_paths='style.css') as interface:
                     outputs=[output_plot, query_rate_plot, radar_plot, ranking_text_display, worst_cases_display, load_trigger]
                 )  
 
-interface.launch()
+interface.launch(share=True)