JaphetHernandez commited on
Commit
5d3dc3e
verified
1 Parent(s): b1382eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -41,7 +41,10 @@ def generate_response(input_text, temperature=0.7, max_new_tokens=100):
41
  outputs = []
42
  for text in streamer:
43
  outputs.append(text)
44
- yield "".join(outputs)
 
 
 
45
 
46
  def main():
47
  st.title("Chat con Meta Llama 3.1 3B")
@@ -61,7 +64,7 @@ def main():
61
  # Generar la respuesta del modelo
62
  if st.button("Generar respuesta"):
63
  with st.spinner("Generando respuesta..."):
64
- response = list(generate_response(initial_prompt))[0] # Obtener la primera respuesta completa
65
  st.write(f"Respuesta del modelo: {response}")
66
 
67
  # Terminar la conversaci贸n
 
41
  outputs = []
42
  for text in streamer:
43
  outputs.append(text)
44
+
45
+ # Devolver la respuesta completa
46
+ return "".join(outputs)
47
+
48
 
49
  def main():
50
  st.title("Chat con Meta Llama 3.1 3B")
 
64
  # Generar la respuesta del modelo
65
  if st.button("Generar respuesta"):
66
  with st.spinner("Generando respuesta..."):
67
+ response = generate_response(initial_prompt) # Obtener la primera respuesta completa
68
  st.write(f"Respuesta del modelo: {response}")
69
 
70
  # Terminar la conversaci贸n