AshenClock commited on
Commit
3215541
·
verified ·
1 Parent(s): 6c53127

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -10
app.py CHANGED
@@ -60,23 +60,19 @@ Regole:
60
  # Funzione per inviare la richiesta al modello Hugging Face
61
  def generate_response(message, max_tokens, temperature):
62
  system_message = create_system_message(rdf_context)
63
- messages = [
64
- {"role": "system", "content": system_message},
65
- {"role": "user", "content": message}
66
- ]
67
 
68
  try:
69
- completion = client.conversational(
 
70
  model="Qwen/Qwen2.5-72B-Instruct",
71
- messages=messages,
72
- max_tokens=max_tokens,
73
  temperature=temperature
74
  )
75
- response = completion.get("generated_text", "").strip()
76
- return response
77
  except Exception as e:
78
  raise HTTPException(status_code=500, detail=f"Errore nell'elaborazione: {str(e)}")
79
-
80
  # Endpoint per generare query SPARQL
81
  @app.post("/generate-query/")
82
  async def generate_query(request: QueryRequest):
 
60
  # Funzione per inviare la richiesta al modello Hugging Face
61
  def generate_response(message, max_tokens, temperature):
62
  system_message = create_system_message(rdf_context)
63
+ full_prompt = f"{system_message}\nUtente: {message}\nAssistente:"
 
 
 
64
 
65
  try:
66
+ # Usa il metodo corretto per invocare il modello Hugging Face
67
+ response = client.text_generation(
68
  model="Qwen/Qwen2.5-72B-Instruct",
69
+ inputs=full_prompt,
70
+ max_new_tokens=max_tokens,
71
  temperature=temperature
72
  )
73
+ return response.strip()
 
74
  except Exception as e:
75
  raise HTTPException(status_code=500, detail=f"Errore nell'elaborazione: {str(e)}")
 
76
  # Endpoint per generare query SPARQL
77
  @app.post("/generate-query/")
78
  async def generate_query(request: QueryRequest):