Update app.py

app.py

@@ -46,13 +46,13 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model_checkpoint = "gokaygokay/Flux-Prompt-Enhance"
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
-enhancer = pipeline('text2text-generation',
-                    model=model,
-                    tokenizer=tokenizer,
-                    repetition_penalty= 1.2,
-                    device=device
-                   )
-enhancer.to(device)
+enhancer = pipeline(
+    'text2text-generation',
+    model=model,
+    tokenizer=tokenizer,
+    repetition_penalty= 1.2,
+    device=device
+)
 max_target_length = 256
 
 @spaces.GPU