lombardata commited on
Commit
9b32559
1 Parent(s): 06809c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ def predict(input_image):
68
  image_processor = AutoImageProcessor.from_pretrained(checkpoint_name)
69
  # predict
70
  inputs = image_processor(input_image, return_tensors="pt")
71
- inputs = inputs
72
  with torch.no_grad():
73
  model_outputs = model(**inputs)
74
  outputs = model_outputs["logits"][0]
 
68
  image_processor = AutoImageProcessor.from_pretrained(checkpoint_name)
69
  # predict
70
  inputs = image_processor(input_image, return_tensors="pt")
71
+ inputs = inputs
72
  with torch.no_grad():
73
  model_outputs = model(**inputs)
74
  outputs = model_outputs["logits"][0]