updating app and utils

app.py

@@ -8,11 +8,13 @@ pipeline.scheduler = DPMSolverMultistepScheduler()
 def infer(text, steps):
     generated_images = pipeline(
         text,
-        batch_size=1,  # Generate one image at a time for each step
+        batch_size=1,
         num_inference_steps=int(steps)
     ).images
 
     return generated_images[0]
 
-demo = gr.Interface(fn=infer, inputs=["text", "number"], outputs="image")
+demo = gr.Interface(fn=infer, inputs=["text", "number"], outputs="image"
+                    description="English text to Chinese glyph. Recommend <10 steps as this space is not running on GPU",
+                    )
 demo.launch()

glyffuser_utils.py

@@ -141,7 +141,8 @@ class GlyffuserPipeline(DiffusionPipeline):
             # 2. compute previous image: x_t -> x_t-1
             image = self.scheduler.step(model_output, t, image, generator=generator, return_dict=False)[0]
 
-        image = (image / 2 + 0.5).clamp(0, 1)
+        # image = (image / 2 + 0.5).clamp(0, 1)
+        image = image.clamp(0, 1)
         image = image.cpu().permute(0, 2, 3, 1).numpy()
         if output_type == "pil":
             image = self.numpy_to_pil(image)