Update app.py

app.py

@@ -112,7 +112,7 @@ def tsaigpt(start:str , max_new_tokens = 300, num_samples =2, tokeniser= token):
     model = GPT(config)
 
     load_checkpoint(fabric, model, checkpoint_path)
-    print(model)
+    #print(model)
     model.eval()
     model.to(device)
     if compile:
@@ -132,10 +132,9 @@ def tsaigpt(start:str , max_new_tokens = 300, num_samples =2, tokeniser= token):
                 output = tokeniser.decode(y[0])
     return output
 
-INTERFACE = gr.Interface(fn=tsaigpt, inputs=[gr.Textbox(label= "Prompt", value= 'All that glisters is not gold.'),
+INTERFACE = gr.Interface(fn=tsaigpt, inputs=[gr.Textbox(label= "Prompt", value= 'We know what we are, but know not what we may be'),
                     gr.Slider(minimum = 300, maximum = 500, value= 300,  label= "Maximum number of tokens to be generated")] ,
                     outputs=gr.Text(label= "Generated Text"), title="TSAI_GPT",
                  description="TSAIGPT is a transformer-based language model with only 0.16 billion parameters, trained on RedPajama 1T Sample.",
-                 examples = [['We know what we are, but know not what we may be',300],
-                ['Sweet are the uses of adversity which, like the toad, ugly and venomous, wears yet a precious jewel in his head',300],]
+                 examples = [['We know what we are, but know not what we may be',300],]
                         ).launch(debug=True)