Spaces:

ayushraj2349-2
/

Automated-Code-Reviewer

Sleeping

ayushraj2349-2 commited on Feb 20

Commit

2e73b5b

verified ·

1 Parent(s): 0040626

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 # ✅ Load the fastest model for code review
 model_name = "EleutherAI/pythia-70m"  # Smallest & fastest alternative
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16).to("cuda")
 # ✅ Function to analyze Python code
 def review_code(code_snippet):

 # ✅ Load the fastest model for code review
 model_name = "EleutherAI/pythia-70m"  # Smallest & fastest alternative
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name).to("cpu")
 # ✅ Function to analyze Python code
 def review_code(code_snippet):