update

app.py

@@ -1,27 +1,16 @@
-import openai
-import os
-from dotenv import load_dotenv
+import gradio as gr
+from transformers import pipeline
 
-# Load environment variables from .env file
-load_dotenv()
-
-# Set the OpenAI API key from environment variable
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Load the GPT-Neo model and tokenizer
+generator = pipeline("text-generation", model="EleutherAI/gpt-neo-125M")
 
 def generate_text(prompt):
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=[
-            {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": prompt},
-        ],
-        max_tokens=150,
-        n=1,
-        stop=None,
-        temperature=0.7,
-    )
-    return response.choices[0].message['content'].strip()
+    # Generate text using GPT-Neo
+    result = generator(prompt, max_length=100, num_return_sequences=1)
+    return result[0]["generated_text"]
+
+# Create a Gradio interface
+iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="GPT-Neo Text Generator")
 
-# Example usage
-prompt = "Once upon a time"
-print(generate_text(prompt))
+# Launch the interface
+iface.launch(share=True)

requirements.txt

@@ -1,4 +1,3 @@
-openai==0.28
-python-dotenv
-gradio  # Include this if you're using Gradio for the interface
-torch   # Include this if you are working with PyTorch models
+transformers
+torch
+gradio