LPX55 commited on
Commit
48e0567
·
1 Parent(s): a677ac6

fix: correct image handling in simple_prediction function and add debug prints for image types

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -173,6 +173,9 @@ register_model_with_metadata(
173
  )
174
 
175
  def preprocess_simple_prediction(image):
 
 
 
176
  # The simple_prediction function expects a PIL image (filepath is handled internally)
177
  return image
178
 
@@ -195,7 +198,7 @@ def simple_prediction(img):
195
  im = load_image(img)
196
 
197
  result = client.predict(
198
- input_image=handle_file(im),
199
  api_name="/simple_predict"
200
  )
201
  return result
@@ -574,7 +577,7 @@ demo = gr.TabbedInterface(
574
  ],
575
  title="Deepfake Detection & Forensics Tools",
576
  theme=None,
577
-
578
  )
579
 
580
  if __name__ == "__main__":
 
173
  )
174
 
175
  def preprocess_simple_prediction(image):
176
+ print(type(image))
177
+ im = load_image(image)
178
+ print(type(im))
179
  # The simple_prediction function expects a PIL image (filepath is handled internally)
180
  return image
181
 
 
198
  im = load_image(img)
199
 
200
  result = client.predict(
201
+ input_image=handle_file(img),
202
  api_name="/simple_predict"
203
  )
204
  return result
 
577
  ],
578
  title="Deepfake Detection & Forensics Tools",
579
  theme=None,
580
+
581
  )
582
 
583
  if __name__ == "__main__":