Skip to content

Commit

Permalink
Update main.py
Browse files Browse the repository at this point in the history
Signed-off-by: Bubbles The Dev <152947339+KernFerm@users.noreply.github.com>
  • Loading branch information
KernFerm authored Sep 3, 2024
1 parent 3b09d47 commit dd82d75
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion main.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ def load_model(model_path='ultralytics/yolov5s'):
print(f"Model loaded in {end_time - start_time:.2f} seconds") # Print the loading time
print("Model loaded and ready.")
print("You can now use the script.")
print("if you have any issues using this join the discord https://discord.fnbubbles420.org/invite HEAD TO 🧑🏫-teaching_tutoring")
return model, model_type # Return the loaded model and model type
except Exception as e: # Catch any exceptions during model loading
print(f"Error loading model: {e}") # Print the error message
Expand Down Expand Up @@ -184,4 +185,4 @@ def main():
cv2.destroyAllWindows()

if __name__ == "__main__":
main()
main()

0 comments on commit dd82d75

Please sign in to comment.