Update app.py

app.py

@@ -38,7 +38,11 @@ def chatbot(input_text, history, model_choice, system_message, max_new_tokens, t
     try:
         response = model_component(payload)  # The response is likely a dictionary
         if isinstance(response, dict) and "choices" in response:
+            # Assuming the response structure is similar to OpenAI's API
             assistant_response = response["choices"][0]["message"]["content"]
+        elif isinstance(response, dict) and "generated_text" in response:
+            # If the response is in a different format, adjust accordingly
+            assistant_response = response["generated_text"]
         else:
             assistant_response = "Unexpected model response format."
     except Exception as e: