Hugging Face's logo

Spaces:
septyandy
/
Rock_Type_Classification
Sleeping

App Files Community
septyandy commited on
Commit
6947e78
·
verified ·
1 Parent(s): e80cc73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from ultralytics import YOLO
3
  # Load model dari Hugging Face
4
- model = YOLO("https://huggingface.co/septyandy/Rock_Type_Classification/resolve/main/best.pt")
5
  # Daftar kelas sesuai model
6
  class_names = ['Igneous_Rock', 'Sedimentary_Rock', 'Metamorphic_Rock']
7
  def classify_image(image):
 
1
  import gradio as gr
2
  from ultralytics import YOLO
3
  # Load model dari Hugging Face
4
+ model = YOLO("https://huggingface.co/septyandy/Klasifikasi_Jenis_Batuan/resolve/main/best_1.pt")
5
  # Daftar kelas sesuai model
6
  class_names = ['Igneous_Rock', 'Sedimentary_Rock', 'Metamorphic_Rock']
7
  def classify_image(image):