Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModel
 
 tokenizer_code2desc = AutoTokenizer.from_pretrained("microsoft/codebert-base")
-model_code2desc = AutoModelForSeq2SeqLM.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)