Nac31 commited on
Commit
0b63cd1
·
1 Parent(s): 7139070
Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -1,25 +1,26 @@
1
  import gradio as gr
2
- from huggingface_hub import InferenceClient
3
  import os
4
  from dotenv import load_dotenv
5
 
6
  load_dotenv()
7
 
8
  # Initialiser le client
9
- client = InferenceClient(
10
- model="Nac31/Sacha-Mistral-0",
11
- token=os.getenv("HF_TOKEN") # Votre token HF
12
  )
13
 
14
  def generate_response(message, temperature=0.7):
15
  try:
16
- response = client.text_generation(
17
- message,
18
- max_new_tokens=500,
19
  temperature=temperature,
20
- do_sample=True
 
21
  )
22
- return response
23
  except Exception as e:
24
  return f"Une erreur s'est produite : {str(e)}"
25
 
 
1
  import gradio as gr
2
+ from openai import OpenAI
3
  import os
4
  from dotenv import load_dotenv
5
 
6
  load_dotenv()
7
 
8
  # Initialiser le client
9
+ client = OpenAI(
10
+ base_url="https://api-inference.huggingface.co/v1",
11
+ api_key=os.getenv('HF_TOKEN')
12
  )
13
 
14
  def generate_response(message, temperature=0.7):
15
  try:
16
+ response = client.chat.completions.create(
17
+ model="Nac31/Sacha-Mistral-0",
18
+ messages=[{"role": "user", "content": message}],
19
  temperature=temperature,
20
+ max_tokens=500,
21
+ stream=False
22
  )
23
+ return response.choices[0].message.content
24
  except Exception as e:
25
  return f"Une erreur s'est produite : {str(e)}"
26