nehulagrawal commited on
Commit
27b3a71
1 Parent(s): e5de768

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ Intrigued? Dive into the world of trading alchemy! 🌌
19
  image_path= [['test/1.jpg', 'foduucom/stockmarket-future-prediction', 640, 0.25, 0.45], ['test/2.jpg', 'foduucom/stockmarket-future-prediction', 640, 0.25, 0.45],['test/3.jpg', 'foduucom/stockmarket-future-prediction', 640, 0.25, 0.45]]
20
 
21
  # Load YOLO model
22
- model = YOLO('foduucom/stockmarket-future-prediction')
 
23
 
24
  #############################################################Image Inference############################################################
25
  def yolov8_img_inference(
 
19
  image_path= [['test/1.jpg', 'foduucom/stockmarket-future-prediction', 640, 0.25, 0.45], ['test/2.jpg', 'foduucom/stockmarket-future-prediction', 640, 0.25, 0.45],['test/3.jpg', 'foduucom/stockmarket-future-prediction', 640, 0.25, 0.45]]
20
 
21
  # Load YOLO model
22
+ model = YOLO.from_pretrained("foduucom/stockmarket-future-prediction")
23
+
24
 
25
  #############################################################Image Inference############################################################
26
  def yolov8_img_inference(