Update app.py

app.py

@@ -2,6 +2,10 @@ import gradio as gr
 from transformers import pipeline
 from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel 
 
+import os
+api_key = os.getenv("OPENAI_API_KEY")
+client = OpenAI(api_key=api_key)
+
 # Load text generation model
 text_generation_model = pipeline("text-generation", model="openai-community/gpt2-large")
 # text_generation_model = pipeline("text-generation", model="distilbert/distilgpt2")