Update app.py

app.py

@@ -1,3 +1,4 @@
+"""
 import torch 
 import tensorflow as tf 
 import flax
@@ -25,3 +26,22 @@ iface = gr.Interface(fn=predict_sentiment, inputs="text", outputs = ["label","nu
 if __name__ == "__main__": 
     iface.launch()
 
+"""
+
+
+from transformers import pipeline AutoModelForCausalLM, AutoTokenizer
+import torch
+
+torch_device = "cuda" if torch.cuda.is_available() else "cpu"
+
+tokenizer = AutoTokenizer.from_pretrained("gpt2")
+
+model = AutoModelForCausalLM.from_pretrained("gpt2", pad_token_id=tokenizer.eos_token_id).to(torch_device)
+
+model_inputs = tokenizer('An explanation of Linear Regression: ', return_tensors='pt').to(torch_device)
+
+output = model.generate(**model_inputs, max_new_tokens=50, do_sample=True, top_p=0.92, top_k=0, temperature=0.6)
+
+print(tokenizer.decode(output[0],skip_special_tokens=True))
+
+