Edit requirements.txt and app.py

app.py

@@ -3,9 +3,14 @@ import numpy as np
 import pandas as pd
 from PIL import Image, ImageDraw, ImageFont
 from ultralytics import YOLO
+from huggingface_hub import hf_hub_download
+from ultralytics import YOLO
+
+# Download model dari Hugging Face
+model_path = hf_hub_download(repo_id="Mnjar/TrashDetection", filename="trash_detection.pt")
 
 # Load YOLO model
-model = YOLO('TrashDetection/trash_detection.pt')
+model = YOLO(model_path)
 
 def predict(image):
     """

requirements.txt

@@ -1,4 +1,5 @@
 gradio
 ultralytics
 pandas
-Pillow
+Pillow
+huggingface_hub