JSenkCC commited on
Commit
f60d968
·
verified ·
1 Parent(s): 4d12a3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -271,8 +271,8 @@ def identify_required_functions(project_path, functionality_description):
271
 
272
 
273
  # Load Hugging Face model and tokenizer
274
- tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-Coder-32B-Instruct")
275
- hf_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-Coder-32B-Instruct", device_map="auto")
276
 
277
  def validate_and_generate_documentation(hf_model, tokenizer, gemini_output, file_contents, functionality_description):
278
  """Uses Hugging Face model to validate functions and generate full documentation."""
 
271
 
272
 
273
  # Load Hugging Face model and tokenizer
274
+ tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen2.5-Coder-14B-Instruct")
275
+ hf_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-Coder-14B-Instruct", device_map="auto")
276
 
277
  def validate_and_generate_documentation(hf_model, tokenizer, gemini_output, file_contents, functionality_description):
278
  """Uses Hugging Face model to validate functions and generate full documentation."""