egmaminta commited on
Commit
2a1e815
·
1 Parent(s): 4555112

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import torch
7
  pipeline = SummarizationPipeline(
8
  model=AutoModelWithLMHead.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune"),
9
  tokenizer=AutoTokenizer.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune", skip_special_tokens=True),
10
- device='cuda')
11
 
12
  def code_summarizer(code):
13
  def code_tokenizer(line):
 
7
  pipeline = SummarizationPipeline(
8
  model=AutoModelWithLMHead.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune"),
9
  tokenizer=AutoTokenizer.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune", skip_special_tokens=True),
10
+ device='cpu')
11
 
12
  def code_summarizer(code):
13
  def code_tokenizer(line):