premalt commited on
Commit
e7a9684
·
1 Parent(s): 24a59b0

migrate to gpt2-medium

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -8,7 +8,7 @@ from typing import List
8
 
9
 
10
  app = FastAPI()
11
- client = InferenceClient("openai-community/gpt2")
12
 
13
  SYSTEM_PROMPT = "You are a very powerful AI to generate interesting stories for short-form content consumption. Make sure to hook the readers attention in the first few seconds. Make sure to be engaging and creative in your responses."
14
 
 
8
 
9
 
10
  app = FastAPI()
11
+ client = InferenceClient("openai-community/gpt2-medium")
12
 
13
  SYSTEM_PROMPT = "You are a very powerful AI to generate interesting stories for short-form content consumption. Make sure to hook the readers attention in the first few seconds. Make sure to be engaging and creative in your responses."
14