Spaces:

julien-c
/

zero-gpu-demo

Sleeping

julien-c HF Staff commited on Nov 24, 2023

Commit

3bf1824

verified ·

1 Parent(s): f636bb5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ import spaces
 import datetime
 import os
 import subprocess
 import gradio as gr
 CUSTOM_CSS = """
@@ -10,6 +11,8 @@ CUSTOM_CSS = """
 }
 """
 @spaces.GPU
 def run_gpu() -> str:
     output: str = ""

 import datetime
 import os
 import subprocess
+import torch
 import gradio as gr
 CUSTOM_CSS = """
 }
 """
+zero = torch.Tensor([0]).cuda()
 @spaces.GPU
 def run_gpu() -> str:
     output: str = ""