yabramuvdi commited on
Commit
43130a6
·
verified ·
1 Parent(s): 9a72c69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
app.py CHANGED
@@ -1,4 +1,18 @@
1
  import os
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  import numpy as np
3
  import torch
4
  import gradio as gr
 
1
  import os
2
+ if os.environ.get("SPACES_ZERO_GPU") is not None:
3
+ import spaces
4
+ else:
5
+ class spaces:
6
+ @staticmethod
7
+ def GPU(func):
8
+ def wrapper(*args, **kwargs):
9
+ return func(*args, **kwargs)
10
+ return wrapper
11
+
12
+ @spaces.GPU
13
+ def fake_gpu():
14
+ pass
15
+
16
  import numpy as np
17
  import torch
18
  import gradio as gr