Update app.py

app.py

@@ -603,7 +603,9 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
                             "model": model_llm
                         }).json()
                         print("RESPONSE TRY", response)
-                        reply = response["choices"][0]["message"]["content"]
+                        # reply = response["choices"][0]["message"]["content"]
+                        reply = response.choices[0].message['content']
+
                         output.append({"Mode": "Image", "type": max_label, "prediction_value": max_rounded_prediction, "content": reply})
                     except:
                         print("DOESN'T WORK")
@@ -645,7 +647,8 @@ def classify(platform, UserInput, Images, Textbox2, Textbox3):
                     "model": model_llm
                 }).json()
     
-                reply = response["choices"][0]["message"]["content"]
+                # reply = response["choices"][0]["message"]["content"]
+                reply = response.choices[0].message['content']
             except:
                 reply = "Maximum messages: 15. Please clear your history and Try Again!"
             output.append({"Mode": "Chat", "content": reply})