zehui127 commited on
Commit
2d377d0
·
1 Parent(s): 1c2dc15
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import re
2
  import gradio as gr
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
-
5
  # Model and tokenizer path on Hugging Face
6
  model_tokenizer_path = "zehui127/Omni-DNA-DNA2Function"
7
 
 
1
  import re
2
  import gradio as gr
3
  from transformers import AutoModelForCausalLM, AutoTokenizer
4
+ import torch
5
  # Model and tokenizer path on Hugging Face
6
  model_tokenizer_path = "zehui127/Omni-DNA-DNA2Function"
7