HMWCS commited on
Commit
0ede587
·
verified ·
1 Parent(s): 3dc7ac0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -11
app.py CHANGED
@@ -1,9 +1,3 @@
1
- import gradio as gr
2
- from PIL import Image
3
- import os
4
- from classifier import GarbageClassifier
5
- from config import Config
6
-
7
  # Check if running in Hugging Face Spaces environment
8
  try:
9
  import spaces
@@ -14,6 +8,13 @@ except ImportError:
14
  HF_SPACES = False
15
  print("Running in local environment")
16
 
 
 
 
 
 
 
 
17
  # Initialize classifier
18
  config = Config()
19
  classifier = GarbageClassifier(config)
@@ -112,8 +113,4 @@ with gr.Blocks(title="Garbage Classification System") as demo:
112
  )
113
 
114
  if __name__ == "__main__":
115
- # Set appropriate launch parameters based on environment
116
- if HF_SPACES:
117
- demo.launch()
118
- else:
119
- demo.launch(share=True)
 
 
 
 
 
 
 
1
  # Check if running in Hugging Face Spaces environment
2
  try:
3
  import spaces
 
8
  HF_SPACES = False
9
  print("Running in local environment")
10
 
11
+ import gradio as gr
12
+ from PIL import Image
13
+ import os
14
+ from classifier import GarbageClassifier
15
+ from config import Config
16
+
17
+
18
  # Initialize classifier
19
  config = Config()
20
  classifier = GarbageClassifier(config)
 
113
  )
114
 
115
  if __name__ == "__main__":
116
+ demo.launch()