Update app.py
Browse files
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-
|
275 |
-
hf_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen2.5-Coder-
|
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."""
|