BhumikaMak commited on
Commit
9e4b6dc
·
verified ·
1 Parent(s): c21abf8

resolved server issue

Browse files
Files changed (1) hide show
  1. app.py +5 -9
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, host='0.0.0.0', port=8080) # Start Netron on port 8080
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.Markdown(
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 link
142
- netron_display = gr.Markdown(
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(