Update app.py

app.py

@@ -1,59 +1,25 @@
 import gradio as gr
-import openai
-import os
+from transformers import pipeline
 
-# Retrieve the OpenAI API key from Hugging Face Secrets
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Load the GPT-2 model from Hugging Face using the pipeline
+generator = pipeline('text-generation', model='gpt2')
 
-# Check if the API key is correctly set
-if openai.api_key is None:
-    raise ValueError("OpenAI API key is not set. Please configure the 'OPENAI_API_KEY' secret in your Hugging Face Space settings.")
-
-def generate_text(prompt, max_length=100, temperature=1.0):
-    try:
-        # Debug: Print the input prompt and settings
-        print(f"Prompt: {prompt}")
-        print(f"Max Length: {max_length}, Temperature: {temperature}")
-
-        # Generate text using OpenAI's GPT-3 (text-davinci-003)
-        response = openai.Completion.create(
-            engine="text-davinci-003",
-            prompt=prompt,
-            max_tokens=max_length,
-            temperature=temperature,
-            n=1,
-            stop=None
-        )
-        
-        # Debug: Print the raw response from OpenAI
-        print(f"API Response: {response}")
-
-        # Extract and return the generated text
-        generated_text = response.choices[0].text.strip()
-        return generated_text
-
-    except openai.error.OpenAIError as e:
-        error_message = f"An OpenAI error occurred: {str(e)}"
-        print(error_message)
-        return error_message
-
-    except Exception as e:
-        error_message = f"An unexpected error occurred: {str(e)}"
-        print(error_message)
-        return error_message
+def generate_text(prompt, max_length=50):
+    # Generate text using the GPT-2 model
+    generated_text = generator(prompt, max_length=max_length, num_return_sequences=1)[0]['generated_text']
+    return generated_text
 
 # Create a Gradio interface
 iface = gr.Interface(
     fn=generate_text,
     inputs=[
         gr.Textbox(lines=3, placeholder="Enter your prompt here...", label="Prompt"),
-        gr.Slider(minimum=50, maximum=500, value=100, label="Max Length"),
-        gr.Slider(minimum=0.1, maximum=1.5, value=1.0, label="Temperature"),
+        gr.Slider(minimum=10, maximum=100, value=50, label="Max Length"),
     ],
     outputs=gr.Textbox(label="Generated Text"),
-    title="GPT-3 Text Generator",
-    description="Enter a prompt and generate text using OpenAI's GPT-3 (text-davinci-003)."
+    title="GPT-2 Text Generator",
+    description="Enter a prompt and generate text using GPT-2 from Hugging Face."
 )
 
-# Launch the interface
+# Launch the Gradio interface
 iface.launch(share=True)