Update app.py

app.py

@@ -7,7 +7,7 @@ import spaces
 # Load the trained model
 
 model_path = 'best.engine'  # Replace with the path to your trained .pt file
-model = YOLO(model_path)
+model = YOLO(model_path,task="classify")
 
 # Define the class indices for figures and tables
 figure_class_index = 3  # class index for figures