Spaces:
Sleeping
Sleeping
spuuntries
commited on
Commit
•
9cb0704
1
Parent(s):
8f7598e
chore: fix deps
Browse files- app.py +1 -1
- requirements.txt +2 -1
app.py
CHANGED
@@ -109,4 +109,4 @@ load_model(best_model, os.path.join(__file__, "..", "bjf8fp.safetensors"))
|
|
109 |
classifier = WasteClassifier(best_model, class_names, device)
|
110 |
|
111 |
demo = interface(classifier)
|
112 |
-
demo.launch(
|
|
|
109 |
classifier = WasteClassifier(best_model, class_names, device)
|
110 |
|
111 |
demo = interface(classifier)
|
112 |
+
demo.launch()
|
requirements.txt
CHANGED
@@ -2,4 +2,5 @@ torch
|
|
2 |
torchvision
|
3 |
pillow
|
4 |
gradio
|
5 |
-
numpy
|
|
|
|
2 |
torchvision
|
3 |
pillow
|
4 |
gradio
|
5 |
+
numpy
|
6 |
+
transformers
|