Spaces:

Junlinh
/

memorability_prediction

Build error

Junlinh commited on Nov 11, 2022

Commit

ae5474f

1 Parent(s): c0b34c9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,6 @@ import torch
 from timm.models import create_model
 def predict(input_img):
     input_img =  Image.fromarray(np.uint8(input_img))
-    print(inpug_img)
     model1 = create_model(
         'resnet50',
         drop_rate=0.5,

 from timm.models import create_model
 def predict(input_img):
     input_img =  Image.fromarray(np.uint8(input_img))
     model1 = create_model(
         'resnet50',
         drop_rate=0.5,