Junlinh commited on
Commit
c0b34c9
·
1 Parent(s): c633e9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import torch
5
  from timm.models import create_model
6
  def predict(input_img):
7
  input_img = Image.fromarray(np.uint8(input_img))
 
8
  model1 = create_model(
9
  'resnet50',
10
  drop_rate=0.5,
 
5
  from timm.models import create_model
6
  def predict(input_img):
7
  input_img = Image.fromarray(np.uint8(input_img))
8
+ print(inpug_img)
9
  model1 = create_model(
10
  'resnet50',
11
  drop_rate=0.5,