[demo] remove limitation and update demo

[felixdittrich92]

This PR:

• removes the limitation so that all models can also be used for possible rotated inputs
• updates the demo and add some more options

Any feedback is welcome 🤗

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (419ea05) 95.77% compared to head (7f18cd1) 95.79%.
Report is 1 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1390      +/-   ##
==========================================
+ Coverage   95.77%   95.79%   +0.02%     
==========================================
  Files         155      155              
  Lines        6951     6942       -9     
==========================================
- Hits         6657     6650       -7     
+ Misses        294      292       -2     

Flag	Coverage Δ
unittests	95.79% <100.00%> (+0.02%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[odulcy-mindee]

This one should be kept and remove the other one ?

[felixT2K]

Mh it's the same arch so i changed the chekpoint because the sec one was already trained on rotated samples so while finetuning it should converge faster as the checkpoint which wasn't trained with rotated samples :)

Makes only sense for TF because the change to antialias=True was much less changing as compared to pytorch

[felixT2K]

In the end we'll train it anyway ^^

[odulcy-mindee]

Thanks !