Update app.py

app.py

@@ -108,15 +108,15 @@ def generate_image(
 
 def gen(
     prompt,
-    width=768,
-    height=1024,
-    scales=5,
-    steps=24,
-    seed=-1,
-    nums=1,
-    lora_scale=1.0, 
-    lora_in="", 
-    lora_add=""
+    width:int=768,
+    height:int=1024,
+    scales:int=5,
+    steps:int=24,
+    seed:int=-1,
+    nums:int=1,
+    lora_scale:float=1.0, 
+    lora_in:str="", 
+    lora_add:str=""
 ):
     enable_lora(lora_scale, lora_in, lora_add)
     generate_image(prompt,width,height,scales,steps,seed,nums)