Update src/app.py

src/app.py

@@ -106,22 +106,17 @@ with gr.Blocks() as demo:
             return gr.update(value=["Optimizer state", "Gradients", "Parameters"])
     
     def change_offloading(evt: gr.SelectData, zero_stage):  # SelectData is a subclass of EventData
-        
         if evt.value == "Optimizer" and evt.selected == False:
-            return gr.CheckboxGroup.update(choices=["Optimizer"], value=[])
-        
+            return gr.update(choices=["Optimizer"], value=[])
         if evt.value == "Optimizer" and evt.selected == True:
             if zero_stage in [1, 2]:
-                return gr.CheckboxGroup.update(choices=["Optimizer"], value=["Optimizer"])
+                return gr.update(choices=["Optimizer"], value=["Optimizer"])
             elif zero_stage == 3:
-                return gr.CheckboxGroup.update(choices=["Optimizer", "Parameters"], value=["Optimizer"])
-
+                return gr.update(choices=["Optimizer", "Parameters"], value=["Optimizer"])
         if evt.value == "Parameters" and evt.selected == False:
-            return gr.CheckboxGroup.update(value=["Optimizer"])
-
+            return gr.update(value=["Optimizer"])
         if evt.value == "Parameters" and evt.selected == True:
-
-            return gr.CheckboxGroup.update(value=["Optimizer", "Parameters"])
+            return gr.update(value=["Optimizer", "Parameters"])