Spaces:

Paarth
/

ForgeT5

Runtime error

Paarth commited on Aug 22, 2022

Commit

d332496

•

1 Parent(s): 0cd5e33

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ warnings.simplefilter('ignore')
 MODEL_NAME = 'Salesforce/codet5-base-multi-sum'
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = SummarizerModel(MODEL_NAME)
-model.load_state_dict(torch.load('/content/drive/MyDrive/PlageBERT/Models/codet5-base-1_epoch-val_loss-0.80.pth'))
 embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
 def summarize(text: str,

 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'))
 embedding_model = SentenceTransformer('all-MiniLM-L6-v2')
 def summarize(text: str,