Update app.py
Browse files
app.py
CHANGED
@@ -17,6 +17,7 @@ from transformers import AutoTokenizer
|
|
17 |
MODEL_NAME = 'Salesforce/codet5-base-multi-sum'
|
18 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
19 |
model = SummarizerModel(MODEL_NAME)
|
|
|
20 |
|
21 |
def summarize(text: str,
|
22 |
tokenizer = tokenizer,
|
@@ -56,5 +57,5 @@ outputs = gr.outputs.Textbox()
|
|
56 |
iface = gr.Interface(fn=summarize,
|
57 |
inputs=['text'],
|
58 |
outputs=outputs,
|
59 |
-
description="
|
60 |
-
iface.launch()
|
|
|
17 |
MODEL_NAME = 'Salesforce/codet5-base-multi-sum'
|
18 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
19 |
model = SummarizerModel(MODEL_NAME)
|
20 |
+
model.load_state_dict(torch.load('codet5-base-1_epoch-val_loss-0.80.pth'))
|
21 |
|
22 |
def summarize(text: str,
|
23 |
tokenizer = tokenizer,
|
|
|
57 |
iface = gr.Interface(fn=summarize,
|
58 |
inputs=['text'],
|
59 |
outputs=outputs,
|
60 |
+
description="Demo for ForgeT5 | Input: A python code | Output: The code summarization")
|
61 |
+
iface.launch(inline = False)
|