Hugging Face's logo

Spaces:
sagarkariya
/
ProjectTestReporting
Sleeping

App Files Community
sagarkariya commited on
Commit
d0174ef
·
verified ·
1 Parent(s): 078b2b8

change model to cpu

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -19,6 +19,8 @@ else:
19
  # Load code T5 model
20
  tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")
21
  model = AutoModelForSeq2SeqLM.from_pretrained("Salesforce/codet5-base")
 
 
22
 
23
 
24
  # 🔍 Function to Generate Responses
 
19
  # Load code T5 model
20
  tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")
21
  model = AutoModelForSeq2SeqLM.from_pretrained("Salesforce/codet5-base")
22
+ device = torch.device("cpu")
23
+ model.to(device)
24
 
25
 
26
  # 🔍 Function to Generate Responses