5Grains commited on
Commit
73e5dcd
·
1 Parent(s): 8fdadcc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ def draw_plot(pred_img, seg):
74
  def sepia(input_img):
75
  input_img = Image.fromarray(input_img)
76
 
77
- inputs = feature_extractor(images=input_img, return_tensors="tf")
78
  outputs = model(**inputs)
79
  logits = outputs.logits
80
 
 
74
  def sepia(input_img):
75
  input_img = Image.fromarray(input_img)
76
 
77
+ inputs = extractor(images=input_img, return_tensors="tf")
78
  outputs = model(**inputs)
79
  logits = outputs.logits
80