Update app.py

app.py

@@ -1,8 +1,7 @@
 import os
 import streamlit as st
-import tensorflow as tf
-import tensorflow_hub as hub
-from transformers import pipeline
+import transformers
+from transformers import AutoModelForCausalLM
 
 os.system("pip freeze > requirements.txt")
 os.system("pip install -r requirements.txt")
@@ -11,9 +10,9 @@ os.system("pip install -r requirements.txt")
 def reverse_prompt_engineer(input_code):
     # Analyze the input code using langchain
     analyzed_code = input_code
-    
+
     # Load the tokenizer and model
-    tokenizer = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
+    tokenizer = AutoModelForCausalLM.from_pretrained("EleutherAI/gpt-neo-1.3B")
     model = tokenizer.model
     
     # Generate a prompt using the analyzed code