Update app.py

app.py

@@ -5,11 +5,14 @@ from pptx.util import Inches
 import streamlit as st
 import subprocess
 import os
-
+try:
+    from transformers import AutoTokenizer, AutoModelForCausalLM
+except ImportError:
+    print("Transformers library is not installed. Please install it using pip install transformers.")
 # Content Generation Agent
 def generate_content(prompt):
-    tokenizer = AutoTokenizer.from_pretrained("EleutherAI/gpt-neox-20b")
-    model = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neox-20b")
+    tokenizer = AutoTokenizer.from_pretrained("gpt2")
+    model = AutoModelForCausalLM.from_pretrained("gpt2")
     inputs = tokenizer.encode(prompt + tokenizer.eos_token, return_tensors='pt')
     outputs = model.generate(inputs, max_length=100, do_sample=True)
     text = tokenizer.decode(outputs[0], skip_special_tokens=True)