Add application file

app.py

@@ -0,0 +1,57 @@
+# -*- coding: utf-8 -*-
+"""Untitled200.ipynb
+
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/drive/1XlVHPlSeue0mvagDZOzdk_wHjLmNlWM1
+"""
+
+#!pip install ultralytics
+#!pip install requests
+#!pip install pillow
+
+from ultralytics import YOLO
+
+# 모델 로드
+model = YOLO("ship.pt")  # 학습된 모델 파일을 로드
+
+# 이미지 추론
+results = model("images.jpeg")  # 이미지 파일로 추론 수행
+
+# 결과 시각화
+results[0].show()  # 첫 번째 결과를 시각적으로 표시
+
+#!pip install gradio
+
+import gradio as gr
+from ultralytics import YOLO
+
+# 1. 모델 초기화
+model = YOLO("ship.pt")
+
+# 2. 추론 함수
+def detect_ship(input_image: str):
+    results = model(input_image)
+    return results[0].plot()[:, :, ::-1]  # BGR to RGB 변환
+
+# 3. Gradio UI
+with gr.Blocks() as demo:
+    with gr.Row():
+        gr.Markdown("## Ship Detection")
+
+    with gr.Row():
+        input_image = gr.Image(label="Input Image", type="filepath")
+        output_image = gr.Image(label="Output", type="numpy")
+
+    with gr.Row():
+        inference_btn = gr.Button("Inference")
+
+    inference_btn.click(fn=detect_ship, inputs=input_image, outputs=output_image)
+
+# 4. Gradio 실행
+demo.launch(inline=False)
+
+# 5. Gradio 종료
+demo.close()
+