1MR commited on
Commit
9c6a84b
·
verified ·
1 Parent(s): 2fc0367

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from fastapi.requests import Request
6
  app = FastAPI()
7
 
8
  # Initialize the text generation pipeline
9
- pipe = pipeline("text2text-generation", model="google/flan-t5-small")
10
 
11
  @app.middleware("http")
12
  async def log_requests(request: Request, call_next):
 
6
  app = FastAPI()
7
 
8
  # Initialize the text generation pipeline
9
+ pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct")
10
 
11
  @app.middleware("http")
12
  async def log_requests(request: Request, call_next):