Spaces:

egmaminta
/

python-code-summarizer

Runtime error

egmaminta commited on Mar 17, 2022

Commit

1cb95a1

1 Parent(s): 1c14dd4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import io
 pipeline = SummarizationPipeline(
     model=AutoModelWithLMHead.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune"),
     tokenizer=AutoTokenizer.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune", skip_special_tokens=True),
-    device=0)
 def code_summarizer(code):
   def code_tokenizer(line):

 pipeline = SummarizationPipeline(
     model=AutoModelWithLMHead.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune"),
     tokenizer=AutoTokenizer.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune", skip_special_tokens=True),
+    device=torch.device('cuda' if torch.cuda.is_available() else 'cpu'))
 def code_summarizer(code):
   def code_tokenizer(line):