MatteoScript commited on
Commit
0ad7c15
·
1 Parent(s): 65e892d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -3
main.py CHANGED
@@ -12,10 +12,10 @@ def format_prompt(message, history):
12
  prompt += f"[INST] {message} [/INST]"
13
  return prompt
14
 
15
- @app.get("/")
16
- def read_root(user_input: str = Query("Come Stai?", description="Input dell'utente")):
17
  history = [] # Puoi definire la history se necessario
18
- generated_response = next(generate(user_input, history)) # Ottieni la risposta generata
19
  return {"response": generated_response} # Restituisci la risposta generata come JSON
20
 
21
  def generate(prompt, history, temperature=0.2, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0):
 
12
  prompt += f"[INST] {message} [/INST]"
13
  return prompt
14
 
15
+ @app.get("/Genera")
16
+ def read_root(input):
17
  history = [] # Puoi definire la history se necessario
18
+ generated_response = next(generate(input, history)) # Ottieni la risposta generata
19
  return {"response": generated_response} # Restituisci la risposta generata come JSON
20
 
21
  def generate(prompt, history, temperature=0.2, max_new_tokens=30000, top_p=0.95, repetition_penalty=1.0):