Spaces:

baglada
/

BOLD

Runtime error

baglada commited on Feb 12, 2023

Commit

602e19f

•

1 Parent(s): cd20b98

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 import csv
 # Load a pre-trained model
-model = transformers.AutoModel.from_pretrained("bert-base-uncased")
 model.eval()
 # Define a function to run the model on input text

 import csv
 # Load a pre-trained model
+model = transformers.AutoModel.from_pretrained("Wessel/DiabloGPT-medium-harrypotter")
 model.eval()
 # Define a function to run the model on input text