Update app.py

app.py

@@ -6,7 +6,7 @@ import os
 
 # Initialize Hugging Face pipelines
 text_generator = pipeline("text-generation", model="gpt2")
-code_generator = pipeline("text2text-generation", model="microsoft/CodeGPT-small-py")
+code_generator = pipeline("text2text-generation", model="t5-base")
 
 # Streamlit App
 st.title("AI Dev Tool Kit")
@@ -22,7 +22,7 @@ if app_mode == "Explorer":
 
 elif app_mode == "In-Chat Terminal":
     st.header("In-Chat Terminal")
-    
+
     def run_terminal_command(command):
         try:
             result = subprocess.run(command, shell=True, capture_output=True, text=True)
@@ -43,7 +43,7 @@ elif app_mode == "In-Chat Terminal":
         terminal_input = gr.Textbox(label="Enter Command or Code")
         terminal_output = gr.Textbox(label="Terminal Output", lines=10)
         terminal_button = gr.Button("Run")
-        
+
         terminal_button.click(
             nlp_code_interpreter,
             inputs=terminal_input,
@@ -63,30 +63,26 @@ elif app_mode == "Tool Box":
 def deploy_to_huggingface(app_name):
     code = f"""
 import gradio as gr
-
 def run_terminal_command(command):
     try:
         result = subprocess.run(command, shell=True, capture_output=True, text=True)
         return result.stdout if result.returncode == 0 else result.stderr
     except Exception as e:
         return str(e)
-
 def nlp_code_interpreter(text):
     response = code_generator(text, max_length=150)
         code = response[0]['generated_text']
         return code, run_terminal_command(code)
-
 with gr.Blocks() as iface:
     terminal_input = gr.Textbox(label="Enter Command or Code")
     terminal_output = gr.Textbox(label="Terminal Output", lines=10)
     terminal_button = gr.Button("Run")
-    
+
     terminal_button.click(
         nlp_code_interpreter,
         inputs=terminal_input,
         outputs=[terminal_output, terminal_output]
     )
-
 iface.launch()
 """