Update app.py

app.py

@@ -1,24 +1,24 @@
 from fastai.vision.all import *
 import gradio as gr
-import torch
-from diffusers import FluxPipeline
-from huggingface_hub import login
-login()
+#import torch
+#from diffusers import FluxPipeline
+#from huggingface_hub import login
+#login()
 
-pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
+#pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
 #pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
 
-prompt = "A cat holding a sign that says hello world"
-image = pipe(
-    prompt,
-    height=1024,
-    width=1024,
-    guidance_scale=3.5,
-    num_inference_steps=50,
-    max_sequence_length=512,
-    generator=torch.Generator("cpu").manual_seed(0)
-).images[0]
-image.save("flux-dev.png")
+#prompt = "A cat holding a sign that says hello world"
+#image = pipe(
+  #  prompt,
+   # height=1024,
+   # width=1024,
+   # guidance_scale=3.5,
+   # num_inference_steps=50,
+    #max_sequence_length=512,
+   # generator=torch.Generator("cpu").manual_seed(0)
+#).images[0]
+#image.save("flux-dev.png")
 
 
 learn = load_learner('export.pkl')