update

inference.py

@@ -42,11 +42,11 @@ def make_image_grid(imgs, rows, cols):
 
 
 def inference_fn(
-        model_id,
-        prompt,
-        num_samples,
-        guidance_scale,
-    ):
+        model_id: str,
+        prompt: str,
+        num_samples: int,
+        guidance_scale: float,
+    ) -> PIL.Image.Image:
 
     # create inference pipeline
     device = 'cuda' if torch.cuda.is_available() else 'cpu'