Update app.py

app.py

@@ -13,5 +13,25 @@ class CFG:
 
 image_gen_model = StableDiffusionPipeline.from_pretrained(
     CFG.image_gen_model_id, torch_dtype=torch.float32,
-    revision="fp16", use_auth_token='hf_'
-)
+    revision="fp16", use_auth_token='hf_pxvzpoafqjfkELFKMLTESNpvmyvkTVuD01' , guidance_scale=9
+)
+
+apply = image_gen_model.to(CFG.device)
+
+def generate_image(prompt):
+    image = apply(
+        prompt, num_inference_steps=CFG.image_gen_steps,
+        generator=CFG.generator , 
+        guidance_scale=CFG.image_gen_guidance_scale
+    ).image[0] 
+    image = image.resize(CFG.image_gen_size)
+    return image
+    )
+
+
+title = "تصوَّر"
+description = "أدخل نص نحولها لك صوره"
+
+iface = gr.interface(fn=generate_image, inputs = "text", outputs = "image" , title= title , description=description)
+iface.launch(
+)