Update app.py

app.py

@@ -47,18 +47,13 @@ def display_image(image, size=None, mode='nearest', unnorm=False, title=''):
     return image
     
 
-mean_latentland = load_model(generator, 'landscape.pt')
-mean_latentface = load_model(generator, 'face.pt')
-mean_latentchurch = load_model(generator, 'church.pt')
+#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
-    elif model_type == 'face':
-        mean_latent = mean_latentface
-    else:
-        mean_latent = mean_latentchurch
+def inferece(num, seed):
+    mean_latent = load_model(generator, 'landscape.pt')
 
     num_im =  int(num)
     random_seed =  int(seed)
@@ -116,7 +111,7 @@ 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.Radio(['church', 'face', 'landscape'], type="value", default="landscape", label="model")
+,gr.inputs.Number(default=90, label="Random Seed"))
 ],
     gr.outputs.Image(type="numpy", label="Output"),
     title=title,