Update app.py

app.py

@@ -17,6 +17,7 @@ from transformers import AutoTokenizer
 MODEL_NAME = 'Salesforce/codet5-base-multi-sum'
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = SummarizerModel(MODEL_NAME)
+model.load_state_dict(torch.load('codet5-base-1_epoch-val_loss-0.80.pth'))
 
 def summarize(text: str, 
               tokenizer = tokenizer,
@@ -56,5 +57,5 @@ outputs = gr.outputs.Textbox()
 iface = gr.Interface(fn=summarize, 
                    inputs=['text'], 
                    outputs=outputs,
-                   description="This is the summarization")
-iface.launch()
+                   description="Demo for ForgeT5 | Input: A python code | Output: The code summarization")
+iface.launch(inline = False)