Add 'precision' parameter for rounding to integer

app.py

@@ -17,8 +17,8 @@ pipe = pipe.to(device)
 
 # define gradio function
 def generate(prompt:str, seed:int, guidance:float, steps:int):
-    generator = torch.Generator(device).manual_seed(int(seed))
-    image = pipe(prompt=prompt, generator=generator, guidance_scale=guidance, num_inference_steps=int(steps)).images[0]
+    generator = torch.Generator(device).manual_seed(seed)
+    image = pipe(prompt=prompt, generator=generator, guidance_scale=guidance, num_inference_steps=steps).images[0]
     return image
 
 if device == "cuda":
@@ -28,9 +28,10 @@ else:
     print("Using CPU.")
 
 # create the gradio UI
+# set precision to 0 to round value to nearest int
 demo = gr.Interface(
     fn=generate,
-    inputs=[gr.Textbox(placeholder="castle on a mountain"), gr.Number(value=123456), gr.Slider(0,10), gr.Number(value=50)],
+    inputs=[gr.Textbox(placeholder="castle on a mountain"), gr.Number(value=123456, precision=0), gr.Slider(0,10), gr.Number(value=50, precision=0)],
     outputs="image",
     allow_flagging="never",
     )