Update app.py

app.py

@@ -4,7 +4,7 @@ import zipfile
 import io
 
 # Load AI assistant (can change model to faster one if needed)
-assistant = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", max_new_tokens=1024)
+assistant = pipeline("text-generation", model="Salesforce/codegen-350M-multi", max_new_tokens=512)
 
 def generate_code(prompt):
     formatted = f"Create a full website. Include:\n[HTML]\n...HTML here...\n[CSS]\n...CSS here...\n[JS]\n...JS here...\nUser Prompt: {prompt}"