spuuntries commited on
Commit
9cb0704
1 Parent(s): 8f7598e

chore: fix deps

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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(share=True)
 
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