Aananda-Giri commited on
Commit
2e985ad
·
1 Parent(s): 02bc66d

Update space

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@
2
  import gradio as gr
3
  import torch
4
  from transformers import PreTrainedTokenizerFast
5
- from gpt_model_code import load_model_n_tokenizer, generate
6
  # Load model and tokenizer once at startup
7
  model, tokenizer = load_model_n_tokenizer()
8
  model.eval()
 
2
  import gradio as gr
3
  import torch
4
  from transformers import PreTrainedTokenizerFast
5
+ from .gpt_model_code import load_model_n_tokenizer, generate
6
  # Load model and tokenizer once at startup
7
  model, tokenizer = load_model_n_tokenizer()
8
  model.eval()
requirements.txt CHANGED
@@ -1,3 +1,3 @@
1
- huggingface_hub==0.25.2
2
  transformers
3
  torch
 
1
+ huggingface_hub
2
  transformers
3
  torch