Hugging Face's logo

Spaces:
ombhojane
/
healthyPlantsApp
Sleeping

App Files Community
ombhojane commited on
Commit
fcfd8b8
·
1 Parent(s): a4429fb

Updates

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
 
3
  from transformers import pipeline
4
 
5
- pipe = pipeline("image-classification", model="ombhojane/healthyPlantsModel")
6
 
7
  def predict_disease(image):
8
  """Predicts the disease of a plant from an image.
 
2
 
3
  from transformers import pipeline
4
 
5
+ pipe = pipeline("image-classification", model="ombhojane/healthyPlantsModel", backend="pytorch")
6
 
7
  def predict_disease(image):
8
  """Predicts the disease of a plant from an image.