Spaces:

ayushraj2349-2
/

Automated-Code-Reviewer

Sleeping

ayushraj2349-2 commited on Feb 21

Commit

ad2a5bc

verified ·

1 Parent(s): eb97e7b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import tempfile  # ✅ Import tempfile to create temp files
 # ✅ Load the fastest model on CPU
-model_name = "EleutherAI/pythia-70m"  # Fastest model for code review
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name).to("cpu")  # Force CPU mode

 import tempfile  # ✅ Import tempfile to create temp files
 # ✅ Load the fastest model on CPU
+model_name = "Salesforce/codegen-350M-mono"  # Fastest model for code review
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name).to("cpu")  # Force CPU mode