SeyedAli commited on
Commit
324f576
·
1 Parent(s): 23e6b97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def FoodClassification(image):
17
 
18
  pipline = pipeline(task="image-classification", model=model)
19
  #output=pipline(model(Image.open(desiredObject), return_tensors='pt'))
20
- output=pipline(image, return_tensors='pt'))
21
  return output
22
 
23
  iface = gr.Interface(fn=FoodClassification, inputs="image", outputs="text")
 
17
 
18
  pipline = pipeline(task="image-classification", model=model)
19
  #output=pipline(model(Image.open(desiredObject), return_tensors='pt'))
20
+ output=pipline(image, return_tensors='pt')
21
  return output
22
 
23
  iface = gr.Interface(fn=FoodClassification, inputs="image", outputs="text")