Update app.py

app.py

@@ -9,8 +9,6 @@ import pickle
 import sys
 import subprocess
 
-from matplotlib import pyplot as plt
-
 import gradio as gr
 import numpy as np
 import torch
@@ -115,11 +113,8 @@ def main():
         #server_port=args.port,
         #share=args.share,
     #)
-    #demo = gr.Interface(fn=image_create, inputs="text", outputs="image")
-    #demo.launch()
-    plt.figure(figsize=(10, 10))
-    plt.imshow(imgs_arr[0])
-    plt.show()
+    demo = gr.Interface(fn=image_create, inputs="text", outputs="image")
+    demo.launch()
 if __name__ == '__main__':
     main()