Update app.py

app.py

@@ -97,15 +97,27 @@ def compute_saliency(input_image, alpha=0.65):
         return blended_image
 
 
+st.title("Visual Saliency Prediction")
 
+col1, col2, col3 = st.columns([1, 1, 1])
 
-with gr.Blocks(head=js_to_prefere_the_back_camera_of_mobilephones()) as demo:
-    with gr.Row():
-        input_image = gr.Image(label="Input Image")
-        output_image = gr.Image(label="Saliency Map")
-        btn = gr.Button("Compute")
-        btn.click(fn=compute_saliency, inputs=input_image, outputs=output_image, api_name="compute_saliency")
+with col1:
+    input_image = st.file_uploader("Upload Input Image", type=["jpg", "jpeg", "png"])
 
-if __name__ == "__main__":
+with col2:
+    output_image = back_camera_input()
+    if image:
+        st.image(image)
 
-    demo.launch()
+with col3:
+    btn = st.button("Compute")
+
+if btn:
+    if input_image is not None:
+        # Perform computation
+        saliency_map = compute_saliency(input_image)
+        
+        # Display output
+        output_image.image(saliency_map, caption="Saliency Map", use_column_width=True)
+    else:
+        st.warning("Please upload an image.")