parrotmaker commited on
Commit
06858da
·
verified ·
1 Parent(s): b8b0e49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import zipfile
4
  import io
5
 
6
  # Load AI assistant (can change model to faster one if needed)
7
- assistant = pipeline("text-generation", model="mistralai/Mixtral-8x7B-Instruct-v0.1", max_new_tokens=1024)
8
 
9
  def generate_code(prompt):
10
  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}"
 
4
  import io
5
 
6
  # Load AI assistant (can change model to faster one if needed)
7
+ assistant = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", max_new_tokens=1024)
8
 
9
  def generate_code(prompt):
10
  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}"