K00B404 commited on
Commit
dd94956
·
verified ·
1 Parent(s): 97c17f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, prompt):
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: