Update app.py

app.py

@@ -1,11 +1,10 @@
 # -*- coding: utf-8 -*-
 
 import gradio as gr
-import openai
-import os
+from transformers import pipeline
 
-# Set your OpenAI API key
-openai.api_key ='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'
+# Load the text generation model (GPT-2)
+text_generator = pipeline('text-generation', model='gpt2')
 
 def get_completion(code_snippet):
     # Code examples to illustrate the output
@@ -44,14 +43,9 @@ def get_completion(code_snippet):
     """
 
     try:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=[{"role": "user", "content": prompt}],
-            max_tokens=300  # Reduced max tokens to limit usage
-        )
-        return response['choices'][0]['message']['content']
-    except openai.error.RateLimitError:
-        return "Quota exceeded. Please try again later."
+        # Generate text using the Hugging Face model
+        response = text_generator(prompt, max_length=300, num_return_sequences=1)
+        return response[0]['generated_text']
     except Exception as e:
         return f"An error occurred: {str(e)}"