Hugging Face's logo

Spaces:
KNDLR
/
trash-ai
Sleeping

App Files Community
KNDLR commited on
Commit
49e4de0
•
1 Parent(s): 622cf48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from model import Pipeline
3
 
4
  pipeline = Pipeline()
5
  labels = {"plastic": "플라스틱", "glass": "유리", "metal": "금속", "paper": "종이", "cardboard": "골판지", "trash": "쓰레기"}
6
- models = {"ResNet152 (Pretrained, 2022-09-02)": "resnet152_0902", "ResNet152 (Pretrained, 2022-08-13)": "resnet152_0813", "ResNet50 (Pretrained, 2022-08-10)": "resnet50_0810"}
7
  is_webcam = False
8
 
9
  def predict(model, file_input, webcam_input):
 
3
 
4
  pipeline = Pipeline()
5
  labels = {"plastic": "플라스틱", "glass": "유리", "metal": "금속", "paper": "종이", "cardboard": "골판지", "trash": "쓰레기"}
6
+ models = {"ResNet152 (Pretrained, latest)": "resnet152_0902", "ResNet152 (Pretrained)": "resnet152_0813", "ResNet50 (Pretrained)": "resnet50_0810"}
7
  is_webcam = False
8
 
9
  def predict(model, file_input, webcam_input):