bambadij commited on
Commit
8d5e03b
·
verified ·
1 Parent(s): 9b6ebea
Files changed (1) hide show
  1. app.py +8 -8
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
- prompt = DEFAULT_PROMPT2 +request.message
 
 
 
 
 
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