Spaces:

chansung
/

paper_qa

Running on CPU Upgrade

chansung commited on Mar 14

Commit

35f930d

•

1 Parent(s): 94c81a2

Update gen/gemini_chat.py

Files changed (1) hide show

gen/gemini_chat.py CHANGED Viewed

@@ -124,6 +124,7 @@ async def gen_text(
     )
     async for resp in resps:
         async for word in _word_generator(resp.text):
             yield word + " "

     )
     async for resp in resps:
+        print(resp.candidate.safety_ratings)
         async for word in _word_generator(resp.text):
             yield word + " "