Spaces:

egmaminta
/

python-code-summarizer

Runtime error

egmaminta commited on Mar 17, 2022

Commit

b6fad16

1 Parent(s): 1cb95a1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,6 +2,7 @@ from transformers import AutoTokenizer, AutoModelWithLMHead, SummarizationPipeli
 import gradio as gr
 import tokenize
 import io
 pipeline = SummarizationPipeline(
     model=AutoModelWithLMHead.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune"),

 import gradio as gr
 import tokenize
 import io
+import torch
 pipeline = SummarizationPipeline(
     model=AutoModelWithLMHead.from_pretrained("SEBIS/code_trans_t5_large_source_code_summarization_python_multitask_finetune"),