Update app.py

app.py

@@ -6,6 +6,7 @@ import logging
 import numpy as np
 import torch
 import datetime
+import gradio as gr
 
 from transformers import (
     CTRLLMHeadModel,
@@ -229,16 +230,19 @@ def main():
             # os.system('clear')
             print(total_sequence)
 
-            fileName = '{}/{}.txt'.format('./songs', date_time)
-            with open(fileName, 'w') as f:
-                f.write(total_sequence)
-
         prompt_text = ""
         if args.prompt:
             break
 
     return generated_sequences
 
+title = "ruGPT3 Song Writer"
+description = "Generate russian songs via fine-tuned ruGPT3"
 
-if __name__ == "__main__":
-    main()
+gr.Interface(
+    process, 
+    gr.inputs.Textbox(lines=1, label="Input text", examples="Как дела? Как дела? Это новый кадиллак"),
+    gr.outputs.Textbox(lines=20, label="Output text"),
+    title=title,
+    description=description,
+    ).launch(enable_queue=True,cache_examples=True)