Update app.py

app.py

@@ -9,7 +9,7 @@ import docker
 from huggingface_hub import HfApi, create_repo
 import importlib
 import os
-from transformers import AutoModelForSequenceClassification, pipeline
+from transformers import pipeline, AutoModelForSequenceClassification
 
 # Initialize Flask app
 app = Flask(__name__)
@@ -119,7 +119,7 @@ plugin_manager.load_plugins()
 
 # AI Assistant
 model = AutoModelForSequenceClassification.from_pretrained("microsoft/CodeGPT-small-py")
-codex_pipeline = pipeline("code-generation", model=model)
+codex_pipeline = pipeline("text-generation", model=model)
 
 hf_api = HfApi()
 
@@ -133,7 +133,7 @@ def generate_app(user_idea, project_name):
 
     # Generate code using Codex
     prompt = f"""Create a simple Streamlit app for the project named '{project_name}'. The app should display the following summary: '{summary}'."""
-    generated_code = codex_pipeline(prompt)[0]['generated_text']
+    generated_code = codex_pipeline(prompt, max_length=516)[0]['generated_text']
 
     # Save the generated code to a file in the project directory
     with open(os.path.join(project_path, "app.py"), "w") as f: