Hugging Face's logo

Spaces:
BeveledCube
/
bevelapi
Sleeping

App Files Community
BeveledCube commited on
Commit
b656ce1
•
1 Parent(s): a89ce24

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +0 -2
main.py CHANGED
@@ -9,8 +9,6 @@ size = "small"
9
  # microsoft/DialoGPT-medium
10
  # microsoft/DialoGPT-large
11
 
12
- os.environ["TRANSFORMERS_CACHE"] = r"./cache"
13
-
14
  # Load the Hugging Face GPT-2 model and tokenizer
15
  model = GPT2LMHeadModel.from_pretrained("microsoft/DialoGPT-medium")
16
  tokenizer = GPT2Tokenizer.from_pretrained("microsoft/DialoGPT-medium")
 
9
  # microsoft/DialoGPT-medium
10
  # microsoft/DialoGPT-large
11
 
 
 
12
  # Load the Hugging Face GPT-2 model and tokenizer
13
  model = GPT2LMHeadModel.from_pretrained("microsoft/DialoGPT-medium")
14
  tokenizer = GPT2Tokenizer.from_pretrained("microsoft/DialoGPT-medium")