Update app.py

app.py

@@ -22,6 +22,9 @@ class SchedulerWrapper:
         except IndexError:
             self._step.step = 0
             return self.scheduler.step(*args, **kwargs)
+    
+    def set_timesteps(self, *args, **kwargs):
+        return self.scheduler.set_timesteps(*args, **kwargs)
 
 # Wrap the scheduler
 model.scheduler = SchedulerWrapper(model.scheduler)