Update app.py

app.py

@@ -33,11 +33,12 @@ JS = """function () {
 }"""
 
 if torch.cuda.is_available():
-    pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16).to(0)
+    pipe = FluxPipeline.from_pretrained(model, torch_dtype=torch.bfloat16)
     pipe.load_lora_weights(default_lora, weight_name = default_weight_name) # default load lora
     pipe.fuse_lora(lora_scale=0.9)
 
 
+
 def scrape_lora_link(url):
     try:
         # Send a GET request to the URL
@@ -100,13 +101,13 @@ def generate_image(
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
     seed = int(seed)
-    print(f'prompt:{prompt}')
     
     text = str(translator.translate(prompt, 'English')) + "," + lora_word
 
+    print(f"Prompt: {text}")
+
     generator = torch.Generator().manual_seed(seed)
     
-
     image = pipe(
         prompt=text,
         height=height,