Update app.py

app.py

@@ -67,7 +67,7 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, model=D
         steps = steps or 16
         guidance = guidance or 3
         
-    model.to(DEVICE)
+    repo.to(DEVICE)
     
     parameters  = {
         "prompt": input,