Spaces:

SandeepU
/

code-explainer-c

Sleeping

SandeepU commited on 6 days ago

Commit

092f4f2

verified ·

1 Parent(s): 2235fa0

Upload model_utils.py

Files changed (1) hide show

model/model_utils.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 def load_model():
     model_name = "Salesforce/codet5-base"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)  # ✅ correct loader
     model.eval()
     device = "cuda" if torch.cuda.is_available() else "cpu"

 def load_model():
     model_name = "Salesforce/codet5-base"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     model.eval()
     device = "cuda" if torch.cuda.is_available() else "cpu"