Update main.py

main.py

@@ -4,20 +4,15 @@ from pydantic import BaseModel
 import uvicorn
 
 
-generator = pipeline('text-generation', model='gpt2')
-set_seed(42)
-
 
 app = FastAPI()
 
-
-
-
 class Item(BaseModel):
     prompt: str
     
 
-
+generator = pipeline('text-generation', model='gpt2')
+set_seed(42)
 
 def generate(item: Item):
     generator(item.prompt, max_length=30, num_return_sequences=5)