Spaces:
Sleeping
Sleeping
update
Browse files
app.py
CHANGED
@@ -22,10 +22,6 @@ SYMPTOMS = [
|
|
22 |
"Nausea", "Headaches", "Breast tenderness", "Acne"
|
23 |
]
|
24 |
|
25 |
-
# Définir les classes de données d'entrée avec Pydantic
|
26 |
-
class RequestData(BaseModel):
|
27 |
-
|
28 |
-
message: str
|
29 |
|
30 |
# Initialiser l'application FastAPI
|
31 |
app = FastAPI()
|
@@ -34,14 +30,18 @@ app = FastAPI()
|
|
34 |
@app.get("/")
|
35 |
async def home():
|
36 |
return 'kitty'
|
37 |
-
|
38 |
@app.post("/analyze")
|
39 |
def predict(request: RequestData):
|
40 |
-
|
|
|
|
|
|
|
|
|
|
|
41 |
try:
|
42 |
completion = client.chat.completions.create(
|
43 |
model="meta/llama-3.1-8b-instruct",
|
44 |
-
messages=prompt,
|
45 |
temperature=0.2,
|
46 |
top_p=0.9,
|
47 |
max_tokens=1024,
|
@@ -56,4 +56,4 @@ def predict(request: RequestData):
|
|
56 |
return {"response": full_response}
|
57 |
|
58 |
except Exception as e:
|
59 |
-
return {"error": f"Erreur : {str(e)}"}
|
|
|
22 |
"Nausea", "Headaches", "Breast tenderness", "Acne"
|
23 |
]
|
24 |
|
|
|
|
|
|
|
|
|
25 |
|
26 |
# Initialiser l'application FastAPI
|
27 |
app = FastAPI()
|
|
|
30 |
@app.get("/")
|
31 |
async def home():
|
32 |
return 'kitty'
|
|
|
33 |
@app.post("/analyze")
|
34 |
def predict(request: RequestData):
|
35 |
+
# Construire la liste de messages
|
36 |
+
prompt = [
|
37 |
+
{"role": "system", "content": DEFAULT_PROMPT2},
|
38 |
+
{"role": "user", "content": request.message}
|
39 |
+
]
|
40 |
+
|
41 |
try:
|
42 |
completion = client.chat.completions.create(
|
43 |
model="meta/llama-3.1-8b-instruct",
|
44 |
+
messages=prompt, # Envoyer la liste de messages
|
45 |
temperature=0.2,
|
46 |
top_p=0.9,
|
47 |
max_tokens=1024,
|
|
|
56 |
return {"response": full_response}
|
57 |
|
58 |
except Exception as e:
|
59 |
+
return {"error": f"Erreur : {str(e)}"}update
|