Update app.py

app.py

@@ -104,7 +104,9 @@ original_sr_load_model = models.sd2_sr.load_model
 def patched_sr_load_model(*args, **kwargs):
     kwargs['device'] = 'cpu'
     model = original_sr_load_model(*args, **kwargs)
-    model.to('cpu')
+    # Check if the model has a to method before calling it
+    if hasattr(model, 'to'):
+        model.to('cpu')
     return model
 models.sd2_sr.load_model = patched_sr_load_model
 
@@ -420,4 +422,4 @@ with gr.Blocks(css=demo_path / 'style.css') as demo:
     )
 
 demo.queue(max_size=20)
-demo.launch(share=True, allowed_paths=[str(TMP_DIR)])
+demo.launch(share=True, allowed_paths=[str(TMP_DIR)])