Ayesha-Majeed commited on
Commit
ccb8d7b
·
verified ·
1 Parent(s): 282b00d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -161,7 +161,7 @@ def extract_fields(image):
161
  # -------------------- Gradio Interface --------------------
162
  demo = gr.Interface(
163
  fn=extract_fields,
164
- inputs=gr.Image(type="pil"),
165
  outputs=[
166
  gr.Text(label="Name"),
167
  gr.Text(label="Email"),
@@ -176,5 +176,12 @@ demo = gr.Interface(
176
  description="Upload a document image to extract structured data fields."
177
  )
178
 
179
- if __name__ == "__main__":
180
- demo.launch()
 
 
 
 
 
 
 
 
161
  # -------------------- Gradio Interface --------------------
162
  demo = gr.Interface(
163
  fn=extract_fields,
164
+ inputs=gr.Image(type="pil", label="Upload Your Document"),
165
  outputs=[
166
  gr.Text(label="Name"),
167
  gr.Text(label="Email"),
 
176
  description="Upload a document image to extract structured data fields."
177
  )
178
 
179
+ # Create a demo block with example image shown after the interface
180
+ with gr.Blocks() as final_ui:
181
+ demo.render()
182
+ gr.Markdown("### 📌 Example Document Format (for reference):")
183
+ gr.Image(value="example_doc.jpeg", interactive=False, label="Sample Document")
184
+
185
+ if __name__ == "__main__":
186
+ final_ui.launch()
187
+