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
- README.md +1 -1
- app.py +4 -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.
|
| 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':
|
| 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':
|
| 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=
|
|
|
|
| 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.
|
| 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
|