Update app.py

app.py

@@ -9,7 +9,7 @@ os.environ["HF_HOME"] = "./cache"  # Store model locally
 MODEL_NAME = "Salesforce/codet5-small"
 
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME, device_map="cpu")
+model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
 
 def generate_code(description, language):
     prompt = f"Generate {language} code: {description}"
@@ -39,6 +39,17 @@ iface = gr.Interface(
         gr.Textbox(lines=5, placeholder="Describe your coding task..."),
         gr.Dropdown(choices=["Python", "JavaScript", "Java"], label="Programming Language")
     ],
-    outputs=[gr.Code(label="Generated Code"), gr.Textbox(label="Execution Output")],
+    outputs=[
+        gr.Code(label="Generated Code"),
+        gr.Textbox(label="Execution Output")
+    ],
     title="Multi-Language Text-to-Code AI",
-    description="Convert natural language descriptions into code in different programming languages!"
+    description="Convert natural language descriptions into code in different programming languages! Run Python code directly in the app.",
+    theme="default",
+    allow_flagging="never",
+    live=True
+)
+
+# Launch the app
+if __name__ == "__main__":
+    iface.launch(share=True, debug=True, server_name="0.0.0.0", server_port=7860)