Update

app.py

@@ -1,25 +1,16 @@
-import os
-import openai
 import gradio as gr
-from dotenv import load_dotenv
+from transformers import pipeline
 
-# Load environment variables from .env file
-load_dotenv()
-
-# Set your OpenAI API key
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Load the FLAN-T5 XXL model
+generator = pipeline("text2text-generation", model="google/flan-t5-xxl")
 
 def generate_text(prompt):
-    response = openai.Completion.create(
-        engine="text-davinci-003",  # Use "gpt-3.5-turbo" or "gpt-4" for newer models
-        prompt=prompt,
-        max_tokens=150,
-        temperature=0.7,
-    )
-    return response.choices[0].text.strip()
+    # Generate text using FLAN-T5 XXL
+    result = generator(prompt, max_length=200, 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-3/4 Text Generator")
+iface = gr.Interface(fn=generate_text, inputs="text", outputs="text", title="FLAN-T5 XXL Text Generator")
 
 # Launch the interface
 iface.launch(share=True)

requirements.txt

@@ -1,3 +1,3 @@
-openai
+transformers
+torch
 gradio
-python-dotenv