Spaces:

nnsohamnn
/

Conv_GPT

Sleeping

nnsohamnn commited on Mar 13

Commit

6cd2458

verified ·

1 Parent(s): dbc7090

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model = TransformerModel(
     num_heads=16,
     dropout=0.1
 )
-model.load_state_dict(torch.load("Conv_GPT_finetuned.pth", map_location=torch.device('cpu')))
 model.eval()

     num_heads=16,
     dropout=0.1
 )
+model.load_state_dict(torch.load("Conv_GPT.pth", map_location=torch.device('cpu')))
 model.eval()