Update app.py

app.py

@@ -1,18 +1,21 @@
+import base64
+import json
+
 import gradio as gr
 import spaces
 import torch
 from spaces.zero.client import _get_token
 
-zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
-
-@spaces.GPU(duration=4*60)
+@spaces.GPU(duration=4*60) # Not possible with IP-based quotas
 def inner():
     return "ok"
 
 def greet(request: gr.Request, n):
     assert inner() ==  "ok"
-    return _get_token(request)
+    token = _get_token(request)
+    payload = token.split('.')[1]
+    payload = f"{payload}{'=' * ((4 - len(payload) % 4) % 4)}"
+    return json.loads(base64.urlsafe_b64decode(payload).decode())
 
-demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.Text())
+demo = gr.Interface(fn=greet, inputs=gr.Number(), outputs=gr.JSON())
 demo.launch()