Update app.py

app.py

@@ -36,7 +36,7 @@ model.load_state_dict(state_dict)
 model.to(device)
 model = torch.compile(model)
 
-def generateText(inputText="JULIET\n", num_tokens=500):
+def generateText(inputText, num_tokens=500):
     start_tokens = enc.encode(inputText)
     # print(start_tokens, len(start_tokens))
     start_tokens = torch.tensor(start_tokens)
@@ -67,19 +67,13 @@ def generateText(inputText="JULIET\n", num_tokens=500):
     decoded = enc.decode(tokens)
     return decoded
 
-title = "GPT from Scratch using char tokenizer to generate text based on training"
-description = "GPT from Scratch using char tokenizer to generate text based on training"
+title = "GPT-2 Trained from Scratch"
+description = "GPT-2 trained on scratch on TinyShakespeare dataset"
 examples = [["ROMEO:\nWith love's light wings did I o'er-perch these walls;\nFor stony limits cannot hold love out,\nAnd what love can do that dares love attempt;\nTherefore thy kinsmen are no let to me.\n", 500],
-            ["ROMEO:\n", 500],
-            ["JULIET:\n", 500],
             ["CAPULET:\nWhy, how now, kinsman! wherefore storm you so?\n", 500],
             ["KING RICHARD II:\nAy, hand from hand, my love, and heart from heart.\nAnd", 500],
-            ["KING RICHARD II:\n", 500],
-            ["CAPULET:\n", 500],
             ["QUEEN:\nBanish us both and send the king with me.\nAnd", 500],
-            ["QUEEN:\n", 500],
-            ["CORIOLANUS:\n", 500],
-            ["MENENIUS:\n", 500]
+            ["CORIOLANUS:\n", 500]
            ]
 
 demo = gr.Interface(