Hugging Face's logo

Spaces:
BeveledCube
/
bevelapi
Running

App Files Community
BeveledCube commited on
Commit
138d9ab
·
verified ·
1 Parent(s): 6b6d0ef

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -9,7 +9,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
9
  import torch
10
 
11
  app = FastAPI()
12
- name = "PygmalionAI/pygmalion-2.7b"
13
 
14
  # microsoft/DialoGPT-small
15
  # microsoft/DialoGPT-medium
@@ -17,7 +17,6 @@ name = "PygmalionAI/pygmalion-2.7b"
17
 
18
  # PygmalionAI/pygmalion-350m
19
  # PygmalionAI/pygmalion-1.3b
20
- # PygmalionAI/pygmalion-2.7b
21
  # PygmalionAI/pygmalion-6b
22
 
23
  # mistralai/Mixtral-8x7B-Instruct-v0.1
 
9
  import torch
10
 
11
  app = FastAPI()
12
+ name = "microsoft/DialoGPT-medium"
13
 
14
  # microsoft/DialoGPT-small
15
  # microsoft/DialoGPT-medium
 
17
 
18
  # PygmalionAI/pygmalion-350m
19
  # PygmalionAI/pygmalion-1.3b
 
20
  # PygmalionAI/pygmalion-6b
21
 
22
  # mistralai/Mixtral-8x7B-Instruct-v0.1