Spaces:

Kvikontent
/

google-gemma-7b

Sleeping

Kvikontent commited on Feb 22

Commit

79feeaa

•

1 Parent(s): f582c6b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ if prompt:
         	"inputs": prompt,
         })
         if output != None:
-            user_msg = st.chat_message("User")
             user_msg.write(prompt)
-            as_msg = st.chat_message("Assistant")
-            as_msg.write(output)

         	"inputs": prompt,
         })
         if output != None:
+            user_msg = cont.chat_message("User")
             user_msg.write(prompt)
+            as_msg = cont.chat_message("Assistant")
+            as_msg.write(output[0]['generated_text'])