ricitos2001 commited on
Commit
8ed406a
·
verified ·
1 Parent(s): 0459b40

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -1,6 +1,4 @@
1
  import time
2
- from typing import Iterable
3
-
4
  import requests
5
  import gradio as gr
6
  from openai import OpenAI
@@ -18,7 +16,7 @@ def obtener_modelos():
18
 
19
  def obtener_respuesta(modelo, api_key, prompt):
20
  client = OpenAI(base_url=URL, api_key=api_key)
21
- mensajes = Iterable["role": "assistant", "content": prompt]
22
  respuesta = client.chat.completions.create(model=modelo, messages=mensajes)
23
  return respuesta.choices[0].message.content
24
 
 
1
  import time
 
 
2
  import requests
3
  import gradio as gr
4
  from openai import OpenAI
 
16
 
17
  def obtener_respuesta(modelo, api_key, prompt):
18
  client = OpenAI(base_url=URL, api_key=api_key)
19
+ mensajes = [{"role": "assistant", "content": prompt}]
20
  respuesta = client.chat.completions.create(model=modelo, messages=mensajes)
21
  return respuesta.choices[0].message.content
22