Spaces:
Runtime error
Runtime error
Damien Benveniste
commited on
Commit
·
5eea5ad
1
Parent(s):
54e091e
latest
Browse files- app.py +17 -2
- requirements.txt +4 -1
app.py
CHANGED
@@ -1,7 +1,22 @@
|
|
1 |
-
from fastapi import FastAPI
|
|
|
|
|
2 |
|
3 |
app = FastAPI()
|
4 |
|
5 |
@app.get("/")
|
6 |
def greet_json():
|
7 |
-
return {"Hello": "World!"}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from fastapi import FastAPI, Request, Response
|
2 |
+
from langserve import APIHandler
|
3 |
+
from langchain_huggingface import HuggingFaceEndpoint
|
4 |
|
5 |
app = FastAPI()
|
6 |
|
7 |
@app.get("/")
|
8 |
def greet_json():
|
9 |
+
return {"Hello": "World!"}
|
10 |
+
|
11 |
+
|
12 |
+
|
13 |
+
llm = HuggingFaceEndpoint(
|
14 |
+
repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
|
15 |
+
)
|
16 |
+
|
17 |
+
api_handler = APIHandler(llm)
|
18 |
+
|
19 |
+
|
20 |
+
@app.get("/stream")
|
21 |
+
async def simple_stream(request: Request):
|
22 |
+
return await api_handler.stream(request)
|
requirements.txt
CHANGED
@@ -1,2 +1,5 @@
|
|
1 |
fastapi
|
2 |
-
uvicorn[standard]
|
|
|
|
|
|
|
|
1 |
fastapi
|
2 |
+
uvicorn[standard]
|
3 |
+
langserve
|
4 |
+
langchain
|
5 |
+
langchain-huggingface
|