Update app.py

app.py

@@ -47,14 +47,24 @@ def display_image(image, size=None, mode='nearest', unnorm=False, title=''):
     return image
     
 
-def inferece(num, seed):
-    model_type = 'landscape' #@param ['church', 'face', 'landscape']
+mean_latentland = load_model(generator, 'landscape.pt')
+mean_latentface = load_model(generator, 'face.pt')
+mean_latentchurch = load_model(generator, 'church.pt')
+
+
+def inferece(num, seed, model_type):
+    if model_type == 'landscape':
+        mean_latent = mean_latentland
+    else if model_type = 'face':
+        mean_latent = mean_latentface
+    else:
+        mean_latent = mean_latentchurch
+
     num_im =  int(num)
     random_seed =  int(seed)
     
     plt.rcParams['figure.dpi'] = 300
     
-    mean_latent = load_model(generator, f'{model_type}.pt')
     
     # pad determines how much of an image is involve in the blending
     pad = 512//4
@@ -106,7 +116,8 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.016
 gr.Interface(
     inferece, 
     [gr.inputs.Number(default=5, label="Number of Images")
-,gr.inputs.Number(default=90, label="Random Seed")],
+,gr.inputs.Number(default=90, label="Random Seed"),gr.inputs.Radio(['church', 'face', 'landscape'], type="value", default="landscape", label="model")
+],
     gr.outputs.Image(type="numpy", label="Output"),
     title=title,
     description=description,