Update app.py

app.py

@@ -49,7 +49,9 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
 enhancer = pipeline('text2text-generation',
                     model=model,
                     tokenizer=tokenizer,
-                    repetition_penalty= 1.2)
+                    repetition_penalty= 1.2,
+                    device=device
+                   )
 enhancer.to(device)
 max_target_length = 256