igmarco commited on
Commit
43d4c63
1 Parent(s): 9fa6241

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  from huggingface_hub import from_pretrained_fastai
2
  import gradio as gr
3
  from fastai.vision.all import *
@@ -10,7 +13,7 @@ labels = learner.dls.vocab
10
 
11
  # Definimos una funci贸n que se encarga de llevar a cabo las predicciones
12
  def predict(img):
13
- img = Image.open(img)
14
  pred,pred_idx,probs = learner.predict(img)
15
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
16
 
 
1
+ import os
2
+ os.system("pip install --upgrade Pillow")
3
+
4
  from huggingface_hub import from_pretrained_fastai
5
  import gradio as gr
6
  from fastai.vision.all import *
 
13
 
14
  # Definimos una funci贸n que se encarga de llevar a cabo las predicciones
15
  def predict(img):
16
+ img = PILImage.create(img)
17
  pred,pred_idx,probs = learner.predict(img)
18
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
19