Spaces:

mamiksik
/

commit-message-autocomplete

Runtime error

mamiksik commited on Jan 21, 2023

Commit

ec8fdf0

•

1 Parent(s): 519c766

Change repository name

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,8 +3,8 @@ import re
 import gradio as gr
 from transformers import RobertaTokenizer, pipeline, AutoModelForMaskedLM
-tokenizer = RobertaTokenizer.from_pretrained("mamiksik/CommitPredictor")
-model = AutoModelForMaskedLM.from_pretrained("mamiksik/CommitPredictor")
 pipe = pipeline("fill-mask", model=model, tokenizer=tokenizer)

 import gradio as gr
 from transformers import RobertaTokenizer, pipeline, AutoModelForMaskedLM
+tokenizer = RobertaTokenizer.from_pretrained("mamiksik/CodeBERTa-commit-message-autocomplete")
+model = AutoModelForMaskedLM.from_pretrained("mamiksik/CodeBERTa-commit-message-autocomplete")
 pipe = pipeline("fill-mask", model=model, tokenizer=tokenizer)