Update app.py

app.py

@@ -1,13 +1,16 @@
+app.py:
 import streamlit as st
 from ultralytics import YOLO
 from PIL import Image
-import io
+import torch
 
 st.set_page_config(layout="centered")
 
 @st.cache_resource
 def load_model():
-    return YOLO("keremberke/yolov8m-chest-xray-classification")
+    model = YOLO('yolov8m.pt')  # Load base YOLOv8 model
+    model.load('keremberke/yolov8m-chest-xray-classification.pt')  # Load weights
+    return model
 
 def main():
     st.title("Analyse Radiographie Thoracique")