Spaces:

Not-Grim-Refer
/

Code-to-Detailed-English-Description

Runtime error

Not-Grim-Refer commited on May 20, 2023

Commit

6b980f5

1 Parent(s): 72e2a36

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 tokenizer_code2desc = AutoTokenizer.from_pretrained("microsoft/codebert-base")
-model_code2desc = AutoModelForSeq2SeqLM.from_pretrained("microsoft/codebert-base")
 def code_to_description(code: str) -> str:
     inputs = tokenizer_code2desc.encode("summarize: " + code, return_tensors="pt", max_length=512)

 import gradio as gr
+from transformers import AutoTokenizer, AutoModel
 tokenizer_code2desc = AutoTokenizer.from_pretrained("microsoft/codebert-base")
+model_code2desc = AutoModel.from_pretrained("microsoft/codebert-base")
 def code_to_description(code: str) -> str:
     inputs = tokenizer_code2desc.encode("summarize: " + code, return_tensors="pt", max_length=512)