Update app.py

app.py

@@ -33,7 +33,7 @@ spectral_sensitivity =  'b'
 TITLE = 'Time-TravelRephotography'
 DESCRIPTION = '''This is an unofficial demo for https://github.com/Time-Travel-Rephotography.
 '''
-ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=hysts.stylegan-human" alt="visitor badge"/></center>'
+ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=Time-TravelRephotography" alt="visitor badge"/></center>'
 
 TOKEN = "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"
 
@@ -94,13 +94,20 @@ def main():
     img_out, _, _ = generator([latent])
     imgs_arr = make_image(img_out)
     iface = gr.Interface(
-      fn=predict, 
-      inputs='text',
-      outputs='text',
-      examples=['result']
+      gr.outputs.Image(type='numpy', label='Output'),
+        title=TITLE,
+        description=DESCRIPTION,
+        article=ARTICLE,
+        theme=args.theme,
+        allow_flagging=args.allow_flagging,
+        live=args.live,
     )
     
-    iface.launch()
+    iface.launch(
+        enable_queue=args.enable_queue,
+        server_port=args.port,
+        share=args.share,
+    )
     
 if __name__ == '__main__':
     main()