Update app.py

app.py

@@ -9,6 +9,8 @@ import pickle
 import sys
 import subprocess
 
+from matplotlib import pyplot as plt
+
 import gradio as gr
 import numpy as np
 import torch
@@ -67,9 +69,12 @@ def load_model(file_name: str, path:str,device: torch.device) -> nn.Module:
         model(z, label, force_fp32=True)
     return model
    
-#def image_create():
+def image_create(input_img):
+    
+
 def greet(name):
     return "Hello " + name + "!"
+    
 def main():
     #torch.cuda.init()
     #if torch.cuda.is_initialized():
@@ -94,14 +99,12 @@ def main():
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)
     #iface = gr.Interface(
-      #fn=predict, 
+      #fn=image_create, 
       #inputs='text',
-      #outputs='text',
-      #examples=['result'],
       #gr.outputs.Image(type='numpy', label='Output'),
-        #title=TITLE,
-        #description=DESCRIPTION,
-        #article=ARTICLE,
+      #title=TITLE,
+      #description=DESCRIPTION,
+      #article=ARTICLE,
         #theme=args.theme,
         #allow_flagging=args.allow_flagging,
         #live=args.live,
@@ -112,8 +115,11 @@ def main():
         #server_port=args.port,
         #share=args.share,
     #)
-    demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-    demo.launch()
+    #demo = gr.Interface(fn=image_create, inputs="text", outputs="image")
+    #demo.launch()
+    plt.figure(figsize=(10, 10))
+    plt.imshow(imgs_arr[0])
+    plt.show()
 if __name__ == '__main__':
     main()