Update app.py

app.py

@@ -33,19 +33,9 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed):
         torch.cuda.empty_cache()
         return image
 
-    if Model == "Club":
-        torch.cuda.max_memory_allocated(device=device)
-        torch.cuda.empty_cache()
-        pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
-        pipe.enable_model_cpu_offload()
-        pipe.vae.enable_slicing()
-        pipe.vae.enable_tiling()
-        image = pipe(prompt=Prompt, guidance_scale=scale, num_images_per_prompt=1, num_inference_steps=steps, width=1024, height=1024,).images[0]
-        return image
-    
     return image
     
-gr.Interface(fn=genie, inputs=[gr.Radio(['PhotoReal', 'Animagine XL 4', "Club"], value='PhotoReal', label='Choose Model'),
+gr.Interface(fn=genie, inputs=[gr.Radio(['PhotoReal', 'Animagine XL 4'], value='PhotoReal', label='Choose Model'),
                                gr.Textbox(label='What you want the AI to generate. 77 Token Limit.'), 
                                gr.Textbox(label='What you Do Not want the AI to generate. 77 Token Limit'),
                                gr.Slider(512, 1024, 768, step=128, label='Height'),