Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -207,7 +207,7 @@ def prepare_input(image, device='cpu'):
|
|
207 |
input_tensor = transform(image).unsqueeze(0).to(device)
|
208 |
return input_tensor
|
209 |
|
210 |
-
def run_inference(image
|
211 |
"""Run inference on a single image"""
|
212 |
global global_model
|
213 |
if global_model is None:
|
|
|
207 |
input_tensor = transform(image).unsqueeze(0).to(device)
|
208 |
return input_tensor
|
209 |
|
210 |
+
def run_inference(image):
|
211 |
"""Run inference on a single image"""
|
212 |
global global_model
|
213 |
if global_model is None:
|