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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -39
app.py CHANGED
@@ -1,39 +0,0 @@
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
- }
20
-
21
- response = requests.post(API_URL, headers=headers, json=payload)
22
- output = response.json()
23
-
24
- if isinstance(output, list) and "generated_text" in output[0]:
25
- return output[0]["generated_text"]
26
- elif "generated_text" in output:
27
- return output["generated_text"]
28
- else:
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()