Update app.py

app.py

@@ -33,47 +33,18 @@ def process_image(image, model_name, inference_engine):
         return f"Error: {str(e)}"
 
 # 创建Gradio界面
-with gr.Blocks(title="Geolocation Reasoning App") as demo:
-    gr.Markdown("# Geolocation Reasoning App")
-    gr.Markdown("Upload an image, and the system will identify the country and region where the image was taken.")
-    
-    with gr.Row():
-        with gr.Column():
-            # 图片上传组件
-            image_input = gr.Image(type="pil", label="Upload Image")
-            
-            # 模型选择
-            model_name = gr.Textbox(
-                label="Model Name",
-                value=DEFAULT_MODEL,
-                info="Input Hugging Face model ID or local model path"
-            )
-            
-            # 推理引擎选择
-            inference_engine = gr.Dropdown(
-                choices=["transformers", "vllm"],
-                value=DEFAULT_ENGINE,
-                label="Inference Engine"
-            )
-            
-            # 提交按钮
-            submit_btn = gr.Button("Start", variant="primary")
-        
-        with gr.Column():
-            # 结果显示
-            output = gr.Textbox(
-                label="Result",
-                lines=10,
-                placeholder="The result will be displayed here..."
-            )
-    
-    # 设置提交按钮的回调函数
-    submit_btn.click(
-        fn=process_image,
-        inputs=[image_input, model_name, inference_engine],
-        outputs=output
-    )
+demo = gr.Interface(
+    fn=process_image,
+    inputs=[
+        gr.Image(type="pil", label="Upload Image"),
+        gr.Textbox(label="Model Name", value=DEFAULT_MODEL),
+        gr.Dropdown(choices=["transformers", "vllm"], value=DEFAULT_ENGINE, label="Inference Engine")
+    ],
+    outputs=gr.Textbox(label="Result"),
+    title="Geolocation Reasoning App",
+    description="Upload an image, and the system will identify the country and region where the image was taken."
+)
 
 # 启动应用
 if __name__ == "__main__":
-    demo.launch() 
+    demo.launch(share=True)