aditya-g07 commited on
Commit
6feecf0
·
1 Parent(s): 867dfa9

Fix runtime errors: Update Gradio version, disable pretrained weights, and fix Spaces configuration

Browse files

- Update Gradio to 4.44.1 to fix JSON schema TypeError
- Set pretrain=False in model configs to avoid missing weights error
- Enable share=True and show_error=True for Hugging Face Spaces
- Update README.md with correct Gradio version

Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +4 -3
  3. requirements.txt +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🔍
4
  colorFrom: blue
5
  colorTo: red
6
  sdk: gradio
7
- sdk_version: 4.44.0
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
 
4
  colorFrom: blue
5
  colorTo: red
6
  sdk: gradio
7
+ sdk_version: 4.44.1
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
app.py CHANGED
@@ -61,7 +61,7 @@ mobilenet_cfg = {
61
  'decay1': 190,
62
  'decay2': 220,
63
  'image_size': 640,
64
- 'pretrain': True,
65
  'return_layers': {'stage1': 1, 'stage2': 2, 'stage3': 3},
66
  'in_channel': 32,
67
  'out_channel': 64
@@ -81,7 +81,7 @@ resnet_cfg = {
81
  'decay1': 70,
82
  'decay2': 90,
83
  'image_size': 840,
84
- 'pretrain': True,
85
  'return_layers': {'layer2': 1, 'layer3': 2, 'layer4': 3},
86
  'in_channel': 256,
87
  'out_channel': 256
@@ -454,5 +454,6 @@ if __name__ == "__main__":
454
  demo.launch(
455
  server_name="0.0.0.0",
456
  server_port=7860,
457
- share=False
 
458
  )
 
61
  'decay1': 190,
62
  'decay2': 220,
63
  'image_size': 640,
64
+ 'pretrain': False, # Don't load pretrained weights
65
  'return_layers': {'stage1': 1, 'stage2': 2, 'stage3': 3},
66
  'in_channel': 32,
67
  'out_channel': 64
 
81
  'decay1': 70,
82
  'decay2': 90,
83
  'image_size': 840,
84
+ 'pretrain': False, # Don't load pretrained weights
85
  'return_layers': {'layer2': 1, 'layer3': 2, 'layer4': 3},
86
  'in_channel': 256,
87
  'out_channel': 256
 
454
  demo.launch(
455
  server_name="0.0.0.0",
456
  server_port=7860,
457
+ share=True,
458
+ show_error=True
459
  )
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- gradio==4.44.0
2
  torch==2.0.1
3
  torchvision==0.15.2
4
  opencv-python==4.8.1.78
 
1
+ gradio==4.44.1
2
  torch==2.0.1
3
  torchvision==0.15.2
4
  opencv-python==4.8.1.78