Hugging Face's logo

Spaces:
TianlaiChen
/
PepMLM
Runtime error

App Files Community
TianlaiChen commited on
Commit
45539e9
•
1 Parent(s): 302efca

gen

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +2 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForMaskedLM
3
  import torch
4
  from torch.distributions.categorical import Categorical
5
-
6
 
7
  # Load the model and tokenizer
8
  tokenizer = AutoTokenizer.from_pretrained("TianlaiChen/PepMLM-650M")
 
2
  from transformers import AutoTokenizer, AutoModelForMaskedLM
3
  import torch
4
  from torch.distributions.categorical import Categorical
5
+ import numpy as np
6
 
7
  # Load the model and tokenizer
8
  tokenizer = AutoTokenizer.from_pretrained("TianlaiChen/PepMLM-650M")
requirements.txt CHANGED
@@ -1,3 +1,4 @@
1
  transformers
2
  gradio
3
- torch
 
 
1
  transformers
2
  gradio
3
+ torch
4
+ numpy