MatteoScript commited on
Commit
b679195
·
verified ·
1 Parent(s): d1d8b1b

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -4
main.py CHANGED
@@ -59,9 +59,8 @@ def LoggaTesto(log_type, data, serializza=True):
59
  print(f"\n{datetime.now()}: ---------------------------------------------------------------| {log_type} |--------------------------------------------------------------\n{formatted_data}")
60
 
61
  def NormalizzaJSON(stringa):
62
- result_data = {"response": stringa}
63
- json_result = json.dumps(result_data)
64
- return json_result
65
 
66
  #--------------------------------------------------- Generazione TESTO ------------------------------------------------------
67
  @app.post("/Genera")
@@ -77,7 +76,7 @@ def generate_text(request: Request, input_data: InputData):
77
  history = []
78
  generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
79
  RispostaJSON = NormalizzaJSON(generated_response)
80
- LoggaTesto("RISPOSTA", RispostaJSON)
81
  return RispostaJSON
82
  else:
83
  input_data.asincrono = False
 
59
  print(f"\n{datetime.now()}: ---------------------------------------------------------------| {log_type} |--------------------------------------------------------------\n{formatted_data}")
60
 
61
  def NormalizzaJSON(stringa):
62
+ json_result = json.dumps(stringa)
63
+ return {"response": json_result}
 
64
 
65
  #--------------------------------------------------- Generazione TESTO ------------------------------------------------------
66
  @app.post("/Genera")
 
76
  history = []
77
  generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
78
  RispostaJSON = NormalizzaJSON(generated_response)
79
+ LoggaTesto("RISPOSTA", RispostaJSON})
80
  return RispostaJSON
81
  else:
82
  input_data.asincrono = False