fireDetection / app.py
ArchiMathur's picture
Update app.py
731f00a verified
raw
history blame
214 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)