gradio installation is moved and a fallback is added

app.py

@@ -668,7 +668,14 @@ def create_comparison_plot(frames_a: list, frames_b: list, prediction_type: str)
 
 
 # Create Gradio interface
-with gr.Blocks(title="RFM Inference Visualizer", theme=gr.themes.Soft()) as demo:
+try:
+    # Try with theme (Gradio 4.0+)
+    demo = gr.Blocks(title="RFM Inference Visualizer", theme=gr.themes.Soft())
+except TypeError:
+    # Fallback for older Gradio versions without theme support
+    demo = gr.Blocks(title="RFM Inference Visualizer")
+
+with demo:
     gr.Markdown(
         """
         # 🤖 RFM (Reward Foundation Model) Inference Visualizer

requirements.txt

@@ -1,7 +1,6 @@
 # Requirements for RFM Eval UI Gradio App
 
 # Core dependencies
-gradio>=4.19.0
 matplotlib>=3.5.0
 numpy>=1.21.0
 torch>=2.0.0
@@ -29,3 +28,5 @@ watchfiles  # For file watching during development
 # For local development, you can also install with: pip install -e ../ (from parent directory)
 git+https://github.com/aliang8/reward_fm.git@93b1ad4b5a530fb32c234bf926b659105e676d00
 
+# Make sure a newer version of gradio is installed
+gradio==4.44.0