Spaces:
Sleeping
Sleeping
resolved server issue
Browse files
app.py
CHANGED
@@ -52,7 +52,7 @@ def process_image(sample_choice, uploaded_image, yolo_versions):
|
|
52 |
# Start Netron backend
|
53 |
def start_netron_backend(model_file):
|
54 |
def serve_netron():
|
55 |
-
netron.start(model_file,
|
56 |
|
57 |
# Launch Netron in a separate thread
|
58 |
threading.Thread(target=serve_netron, daemon=True).start()
|
@@ -70,7 +70,7 @@ def start_netron_backend(model_file):
|
|
70 |
return False
|
71 |
|
72 |
# Check server readiness
|
73 |
-
wait_for_netron("http://localhost:8080/")
|
74 |
|
75 |
# View Netron model
|
76 |
def view_netron_model():
|
@@ -80,9 +80,7 @@ def view_netron_model():
|
|
80 |
|
81 |
# Start Netron backend
|
82 |
start_netron_backend(preloaded_model_file)
|
83 |
-
return gr.
|
84 |
-
"### [View Model in Netron](http://localhost:8080/) - Click the link if Netron does not load properly."
|
85 |
-
)
|
86 |
|
87 |
# Custom CSS for styling (optional)
|
88 |
custom_css = """
|
@@ -138,10 +136,8 @@ with gr.Blocks(css=custom_css) as interface:
|
|
138 |
height=500,
|
139 |
)
|
140 |
|
141 |
-
# Display Netron
|
142 |
-
netron_display =
|
143 |
-
"### [View Model in Netron](http://localhost:8080/) - Click the link if Netron does not load properly."
|
144 |
-
)
|
145 |
|
146 |
# Sample selection update
|
147 |
sample_selection.change(
|
|
|
52 |
# Start Netron backend
|
53 |
def start_netron_backend(model_file):
|
54 |
def serve_netron():
|
55 |
+
netron.start(model_file, address='0.0.0.0', port=8080, browse=False) # Start Netron without browser
|
56 |
|
57 |
# Launch Netron in a separate thread
|
58 |
threading.Thread(target=serve_netron, daemon=True).start()
|
|
|
70 |
return False
|
71 |
|
72 |
# Check server readiness
|
73 |
+
wait_for_netron("http://localhost:8080/", timeout=15)
|
74 |
|
75 |
# View Netron model
|
76 |
def view_netron_model():
|
|
|
80 |
|
81 |
# Start Netron backend
|
82 |
start_netron_backend(preloaded_model_file)
|
83 |
+
return gr.HTML('<iframe src="http://localhost:8080/" width="100%" height="600px"></iframe>')
|
|
|
|
|
84 |
|
85 |
# Custom CSS for styling (optional)
|
86 |
custom_css = """
|
|
|
136 |
height=500,
|
137 |
)
|
138 |
|
139 |
+
# Display Netron iframe
|
140 |
+
netron_display = view_netron_model()
|
|
|
|
|
141 |
|
142 |
# Sample selection update
|
143 |
sample_selection.change(
|