Update main.py

main.py

@@ -29,6 +29,22 @@ app.mount("/static", StaticFiles(directory="static"), name="static")
 # Tell Gradio which paths are allowed to be served
 os.environ["GRADIO_ALLOWED_PATHS"] = str(static_dir.resolve())
 
+
+token = _get_token(gr.Request)
+payload = token.split('.')[1]
+payload = f"{payload}{'=' * ((4 - len(payload) % 4) % 4)}"
+payload = json.loads(base64.urlsafe_b64decode(payload).decode())
+print(f"Token payload outside of function: {payload}")  # For debugging
+
+
+@spaces.GPU(1)  # Specify GPU duration in seconds, added for testing
+def init_gpu(text):
+    """Example GPU function - in reality, this might be model inference"""
+    time.sleep(10)
+
+init_gpu()
+
+
 @spaces.GPU(duration=4*60)  # Specify GPU duration in seconds, added for testing
 def process_text(text):
     """Example GPU function - in reality, this might be model inference"""