backend integration + dynamic nodes

shuja-shah-dev · Jun 12, 2024 · c130d81 · c130d81
1 parent cb3e130
commit c130d81
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/flask/app.py b/flask/app.py
@@ -17,8 +17,7 @@
 app = Flask(__name__)
 CORS(app, origins=['*'])
 
-model = YOLO('./model.pt')
-model2 = YOLO('./yolov8n.pt')
+
 
 def rotation(image):
     model = load_model('orientation_model.h5')