Spaces:
Running
on
Zero
Running
on
Zero
After ZeroGPU was assigned
Browse files
app.py
CHANGED
@@ -189,7 +189,7 @@ model = None
|
|
189 |
# # return model(**data)
|
190 |
# return model(str(data))
|
191 |
|
192 |
-
|
193 |
def estimate(data):
|
194 |
global model
|
195 |
with torch.no_grad():
|
|
|
189 |
# # return model(**data)
|
190 |
# return model(str(data))
|
191 |
|
192 |
+
@spaces.GPU(duration=30)
|
193 |
def estimate(data):
|
194 |
global model
|
195 |
with torch.no_grad():
|