Update app.py

app.py

@@ -13,7 +13,10 @@ def generate_dialogue(character_dropdown):
     return decode(newmodel.model.generate(context, max_new_tokens=100)[0].tolist())
   else:
     context = torch.tensor([encode(character_dropdown)], dtype=torch.long)
-    return decode(newmodel.model.generate(context, max_new_tokens=100)[0].tolist())
+    output_dialogue = decode(newmodel.model.generate(context, max_new_tokens=100)[0].tolist())
+    # remove extra dialogue returned
+    output_dialogue = str(output_dialogue.split("\n\n")[0])
+    return output_dialogue
 
       
 
@@ -72,6 +75,7 @@ HTML_TEMPLATE = """
     <!-- Content -->
     <h1>SHAKESPEARE  DIALOGUE  GENERATOR</h1>
     <p>Generate dialogue for Shakespearean character by selecting character from dropdown.</p>
+    <p>Model: GPT, Dataset: Tiny Shakespeare, Token limit: 100.</p>
 """
 
 with gr.Blocks(theme=gr.themes.Glass(),css=".gradio-container {background: url('file=https://github.com/Delve-ERAV1/S20/assets/11761529/c0ff84a4-dde6-473e-a820-d3797040eb9d')}") as interface: