Update main.py

main.py

@@ -2,6 +2,9 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from huggingface_hub import InferenceClient
 import uvicorn
+from transformers import pipeline
+
+model_name = "distilgpt2"
 
 
 app = FastAPI()