fcakyon commited on
Commit
e271ca3
1 Parent(s): 4765a1d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -11,9 +11,7 @@ import torch
11
 
12
  IMAGE_SIZE = 640
13
 
14
- model_path = hf_hub_download(
15
- "kadirnar/deprem_model_v1", filename="last.pt", revision="main"
16
- )
17
 
18
 
19
  current_device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -21,7 +19,7 @@ model_types = ["YOLOv5", "YOLOv5 + SAHI"]
21
  # Model
22
  model = AutoDetectionModel.from_pretrained(
23
  model_type="yolov5",
24
- model_path=model_path,
25
  device=current_device,
26
  confidence_threshold=0.5,
27
  image_size=IMAGE_SIZE,
 
11
 
12
  IMAGE_SIZE = 640
13
 
14
+ model_id = 'deprem-ml/Binafarktespit-yolo5x-v1-xview'
 
 
15
 
16
 
17
  current_device = "cuda" if torch.cuda.is_available() else "cpu"
 
19
  # Model
20
  model = AutoDetectionModel.from_pretrained(
21
  model_type="yolov5",
22
+ model_path=model_id,
23
  device=current_device,
24
  confidence_threshold=0.5,
25
  image_size=IMAGE_SIZE,