kookoobau commited on
Commit
09d7e31
·
1 Parent(s): c0b6ea0
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from huggingface_hub import hf_hub_download
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
  import gradio as gr
9
 
10
- model_path = hf_hub_download(repo_id="microsoft/DialoGPT-medium", repo_type="model", use_auth_token=None, filename="pytorch_model.bin")
11
  # Load the tokenizer and model
12
  tokenizer = AutoTokenizer.from_pretrained(model_path)
13
  model = AutoModelForCausalLM.from_pretrained(model_path)
 
7
  from transformers import AutoTokenizer, AutoModelForCausalLM
8
  import gradio as gr
9
 
10
+ model_path = hf_hub_download(repo_id="microsoft/DialoGPT-medium", filename="pytorch_model.bin")
11
  # Load the tokenizer and model
12
  tokenizer = AutoTokenizer.from_pretrained(model_path)
13
  model = AutoModelForCausalLM.from_pretrained(model_path)