cleaned the code

app.py

@@ -82,18 +82,15 @@ def predict(prompt):
         stream=True
     )
 
-    partialMessage = ""
     chunkCount = 0
     for chunk in response:
         chunkContent = chunk.choices[0].delta.content
         if chunkContent:
             chunkCount += 1
             yield chunkContent
-    return partialMessage
 
 
 st.title("Chat with Aariz baby 👶🏻")
-# st.image("baby.png")
 
 if "messages" not in st.session_state:
     st.session_state.messages = []