Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from transformers import GPT2Tokenizer, GPT2LMHeadModel
|
|
6 |
#generator = pipeline('text-generation', model="microsoft/DialoGPT-medium")
|
7 |
tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
|
8 |
#model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
|
9 |
-
model = GPT2LMHeadModel.from_pretrained('zmbfeng/FineTune-1')
|
10 |
|
11 |
def create_response(input_str):
|
12 |
#output_raw= generator(input_str)
|
|
|
6 |
#generator = pipeline('text-generation', model="microsoft/DialoGPT-medium")
|
7 |
tokenizer = GPT2Tokenizer.from_pretrained('microsoft/DialoGPT-medium')
|
8 |
#model = GPT2LMHeadModel.from_pretrained('microsoft/DialoGPT-medium')
|
9 |
+
model = GPT2LMHeadModel.from_pretrained('zmbfeng/FineTune-1',use_auth_token=True)
|
10 |
|
11 |
def create_response(input_str):
|
12 |
#output_raw= generator(input_str)
|