PierreH commited on
Commit
60d9eed
·
verified ·
1 Parent(s): 3005c99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -19
app.py CHANGED
@@ -1,27 +1,19 @@
 
1
  import gradio as gr
2
  import requests
3
- import os
4
 
5
  API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
6
- HF_TOKEN = os.environ["HF_TOKEN"] # sécurisé via secret
7
-
8
  headers = {"Authorization": f"Bearer {HF_TOKEN}"}
9
 
10
- def generate_suggestion(mots):
11
- prompt = f"""
12
- Tu es un thérapeute-poète.
13
- Voici quelques mots sensoriels : {mots}.
14
- Rédige une suggestion hypnotique de 5 phrases.
15
- Chaque phrase doit évoquer une sensation corporelle ou symbolique, dans un style poétique, immersif et doux.
16
- Termine toujours par •°○
17
- """
18
-
19
  payload = {
20
  "inputs": prompt,
21
  "parameters": {
22
- "max_new_tokens": 160,
23
- "temperature": 0.85,
24
- "top_p": 0.9,
25
  "do_sample": True
26
  }
27
  }
@@ -37,11 +29,11 @@ Termine toujours par •°○
37
  return "⚠️ Erreur : aucune réponse utile."
38
 
39
  iface = gr.Interface(
40
- fn=generate_suggestion,
41
- inputs=gr.Textbox(label="Mots sensoriels (séparés par virgule)", lines=2, placeholder="souffle, lumière, peau..."),
42
  outputs="text",
43
- title="Résonance Hypnopoétique •°○",
44
- description="Génère une suggestion poétique et sensorielle en 5 phrases, inspirée de vos mots-clés."
45
  )
46
 
47
  iface.launch()
 
1
+ import os
2
  import gradio as gr
3
  import requests
 
4
 
5
  API_URL = "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1"
6
+ HF_TOKEN = os.environ["HF_TOKEN"]
 
7
  headers = {"Authorization": f"Bearer {HF_TOKEN}"}
8
 
9
+ # Fonction qui reçoit le prompt complet
10
+ def generate_from_prompt(prompt):
 
 
 
 
 
 
 
11
  payload = {
12
  "inputs": prompt,
13
  "parameters": {
14
+ "max_new_tokens": 180,
15
+ "temperature": 0.9,
16
+ "top_p": 0.95,
17
  "do_sample": True
18
  }
19
  }
 
29
  return "⚠️ Erreur : aucune réponse utile."
30
 
31
  iface = gr.Interface(
32
+ fn=generate_from_prompt,
33
+ inputs=gr.Textbox(label="Prompt complet", lines=10),
34
  outputs="text",
35
+ title="Prompt direct pour IA hypnopoétique",
36
+ description="Utilisable avec un backend (server.js) qui envoie un prompt complet."
37
  )
38
 
39
  iface.launch()