Update main.py (#1)
Browse files- Update main.py (1081fbebb6b53e9903e2ce479f5178095f353d22)
main.py
CHANGED
@@ -9,7 +9,7 @@ app = FastAPI()
|
|
9 |
|
10 |
#generator = pipeline('text-generation',model='gpt2')
|
11 |
#generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
|
12 |
-
generator = pipeline("text-generation", model="lmsys/vicuna-7b-v1.5")
|
13 |
|
14 |
|
15 |
@app.get("/")
|
|
|
9 |
|
10 |
#generator = pipeline('text-generation',model='gpt2')
|
11 |
#generator = pipeline('text-generation',model='Open-Orca/Mistral-7B-OpenOrca')
|
12 |
+
#generator = pipeline("text-generation", model="lmsys/vicuna-7b-v1.5")
|
13 |
|
14 |
|
15 |
@app.get("/")
|