Update main.py
#2
by
awakenai
- opened
main.py
CHANGED
|
@@ -7,7 +7,8 @@ from fastapi import FastAPI
|
|
| 7 |
|
| 8 |
app = FastAPI()
|
| 9 |
|
| 10 |
-
generator = pipeline('text-generation',model='gpt2')
|
|
|
|
| 11 |
|
| 12 |
@app.get("/")
|
| 13 |
async def root():
|
|
|
|
| 7 |
|
| 8 |
app = FastAPI()
|
| 9 |
|
| 10 |
+
#generator = pipeline('text-generation',model='gpt2')
|
| 11 |
+
generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
|
| 12 |
|
| 13 |
@app.get("/")
|
| 14 |
async def root():
|