airworkx commited on
Commit
23c33b8
1 Parent(s): d50619c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import google.generativeai as genai
2
  import streamlit as st
3
  import time
4
  import random
5
- from utils import SAFETY_SETTINGS
6
 
7
 
8
  st.set_page_config(
@@ -54,7 +54,7 @@ if "app_key" in st.session_state:
54
  message_placeholder.markdown("Thinking...")
55
  try:
56
  full_response = ""
57
- for chunk in chat.send_message(prompt, stream=True, safety_settings = SAFETY_SETTINGS):
58
  word_count = 0
59
  random_int = random.randint(5, 10)
60
  for word in chunk.text:
 
2
  import streamlit as st
3
  import time
4
  import random
5
+ from utils import safety_settings
6
 
7
 
8
  st.set_page_config(
 
54
  message_placeholder.markdown("Thinking...")
55
  try:
56
  full_response = ""
57
+ for chunk in chat.send_message(prompt, stream=True, safety_settings = safety_settings):
58
  word_count = 0
59
  random_int = random.randint(5, 10)
60
  for word in chunk.text: