Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -42,10 +42,10 @@ VOYEUR_SPECIALIST_CONFIG = {
|
|
| 42 |
],
|
| 43 |
}
|
| 44 |
|
| 45 |
-
#
|
| 46 |
client = SambaNova(
|
| 47 |
-
api_key=os.getenv("SAMBANOVA_API_KEY"),
|
| 48 |
-
base_url=os.getenv("SAMBANOVA_BASE_URL")
|
| 49 |
)
|
| 50 |
|
| 51 |
# ============================================
|
|
@@ -91,12 +91,13 @@ def build_prompt(user_text: str) -> str:
|
|
| 91 |
return SYSTEM_PROMPT_BASE + "\n\nUSER:\n" + user_text
|
| 92 |
|
| 93 |
def call_model(prompt: str) -> str:
|
| 94 |
-
# Llamada al modelo SambaNova para obtener la respuesta real
|
| 95 |
response = client.chat.completions.create(
|
| 96 |
model=MODEL_NAME,
|
| 97 |
-
messages=[
|
|
|
|
|
|
|
|
|
|
| 98 |
)
|
| 99 |
-
# Extraer el texto de la respuesta
|
| 100 |
return response.choices[0].message["content"]
|
| 101 |
|
| 102 |
def infer(user_text: str) -> str:
|
|
@@ -197,12 +198,11 @@ with gr.Blocks(title="Voyeur Prompt Generator") as demo:
|
|
| 197 |
out = gr.Textbox(
|
| 198 |
label="Prompt generado (Ingl茅s)",
|
| 199 |
lines=8,
|
| 200 |
-
|
| 201 |
)
|
| 202 |
|
| 203 |
btn.click(fn=infer, inputs=inp, outputs=out)
|
| 204 |
|
| 205 |
-
|
| 206 |
if __name__ == "__main__":
|
| 207 |
demo.launch()
|
| 208 |
|
|
|
|
| 42 |
],
|
| 43 |
}
|
| 44 |
|
| 45 |
+
# Cliente SambaNova inicializado con variables de entorno
|
| 46 |
client = SambaNova(
|
| 47 |
+
api_key=os.getenv("SAMBANOVA_API_KEY"),
|
| 48 |
+
base_url=os.getenv("SAMBANOVA_BASE_URL")
|
| 49 |
)
|
| 50 |
|
| 51 |
# ============================================
|
|
|
|
| 91 |
return SYSTEM_PROMPT_BASE + "\n\nUSER:\n" + user_text
|
| 92 |
|
| 93 |
def call_model(prompt: str) -> str:
|
|
|
|
| 94 |
response = client.chat.completions.create(
|
| 95 |
model=MODEL_NAME,
|
| 96 |
+
messages=[
|
| 97 |
+
{"role": "system", "content": SYSTEM_PROMPT_BASE},
|
| 98 |
+
{"role": "user", "content": prompt}
|
| 99 |
+
],
|
| 100 |
)
|
|
|
|
| 101 |
return response.choices[0].message["content"]
|
| 102 |
|
| 103 |
def infer(user_text: str) -> str:
|
|
|
|
| 198 |
out = gr.Textbox(
|
| 199 |
label="Prompt generado (Ingl茅s)",
|
| 200 |
lines=8,
|
| 201 |
+
buttons=["copy"]
|
| 202 |
)
|
| 203 |
|
| 204 |
btn.click(fn=infer, inputs=inp, outputs=out)
|
| 205 |
|
|
|
|
| 206 |
if __name__ == "__main__":
|
| 207 |
demo.launch()
|
| 208 |
|