Hugging Face's logo

Spaces:
V8055
/
Leaf_Care
Sleeping

App Files Community
V8055 commited on
Commit
3efdae4
·
verified ·
1 Parent(s): d30d334

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -2,8 +2,7 @@ import gradio as gr
2
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
3
  from PIL import Image
4
  import torch
5
- import json
6
- class_map = json.load(open("class_map.json"))
7
  # Load model and feature extractor
8
  model_name = "yusuf802/Leaf-Disease-Predictor"
9
  extractor = AutoFeatureExtractor.from_pretrained(model_name)
 
2
  from transformers import AutoFeatureExtractor, AutoModelForImageClassification
3
  from PIL import Image
4
  import torch
5
+
 
6
  # Load model and feature extractor
7
  model_name = "yusuf802/Leaf-Disease-Predictor"
8
  extractor = AutoFeatureExtractor.from_pretrained(model_name)