fireDetection / app.py
ArchiMathur's picture
Update app.py
defef4e verified
Raw
History Blame
212 Bytes
from ultralytics import YOLO
import streamlit as st
import numpy as np
from PIL import Image
model=YOLO("best.pt")
model.predict(source="360_F_627393178_jEEMrrK3XTqw8Ib9pM4jQP6KnHTgS5qL.jpg",imgsz=640,save=True)