bashyaldhiraj2067 commited on
Commit
aa65d0c
·
verified ·
1 Parent(s): b8930e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -160,7 +160,7 @@ class TransformerCopyHF(PreTrainedModel):
160
  def forward(self, input_ids, attention_mask=None, labels=None):
161
  return self.model(input_ids, attention_mask, labels)
162
 
163
- model = TransformerCopyHF.from_pretrained("bashyaldhiraj2067/remove1_copy_transformer")
164
  model.eval()
165
 
166
  # =====================
 
160
  def forward(self, input_ids, attention_mask=None, labels=None):
161
  return self.model(input_ids, attention_mask, labels)
162
 
163
+ model = TransformerCopyHF.from_pretrained("bashyaldhiraj2067/remove_copy_transformer")
164
  model.eval()
165
 
166
  # =====================