Update app.py

app.py

@@ -43,15 +43,20 @@ def gemini_predict(prompt):
                         Query : {prompt}
                         """
     response = gemini_flash.generate_content(system_message)
-    return(response.text)
+    return(str(response.text)[-2])
 
 
 MAX_SEED = np.iinfo(np.int32).max
 DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"
 
+#Setting up Flux (Schnell) Inpainting, with Realism LoRA weights
 inpaint_pipe = FluxInpaintPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
-inpaint_pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v1')
+inpaint_pipe.load_lora_weights('hugovntr/flux-schnell-realism', weight_name='schnell-realism_v2.3.safetensors', adapter_name="better")
+inpaint_pipe.set_adapters(["better"], adapter_weights=[1.0])
+inpaint_pipe.fuse_lora(adapter_name=["better"], lora_scale=1.0)
+inpaint_pipe.unload_lora_weights()
 
+torch.cuda.empty_cache()
 
 @spaces.GPU()
 def process(input_image_editor, mask_image, input_text, strength, seed, randomize_seed, num_inference_steps, guidance_scale=3.5, progress=gr.Progress(track_tqdm=True)):
@@ -133,11 +138,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 type='pil', image_mode='RGB', label='Generated Mask')
             with gr.Accordion("Debug Info", open=False):
                 output_seed = gr.Number(label="Used Seed")
+                gr.Textbox(gemini_predict(prompt))
 
     submit_button_component.click(
         fn=process,
         inputs=[input_image_component, uploaded_mask_component, input_text_component, strength_slider, seed_number, randomize_seed, num_inference_steps, guidance_scale],
-        outputs=[output_image_component, output_mask_component, output_seed]
+        outputs=[output_image_component, output_mask_component, output_seed, identified_item]
     )
 
 demo.launch(debug=False, show_error=True)