hh871 commited on
Commit
8be5261
1 Parent(s): dc94328

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -11,6 +11,7 @@ learn = load_learner('movie_score_regression_resnet18.pkl')
11
  def predict(img):
12
  img = PILImage.create(img)
13
  pred, _, _= learn.predict(img)
 
14
  return str(round(pred[0], 1))
15
 
16
  gr.Interface(
11
  def predict(img):
12
  img = PILImage.create(img)
13
  pred, _, _= learn.predict(img)
14
+ print(pred)
15
  return str(round(pred[0], 1))
16
 
17
  gr.Interface(