Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 import models
 import time
     
-text = "<h1 style='text-align: center; color: blue; font-size: 30px;'>TCO Comparison Calculator"
-text1 = "<h1 style='text-align: center; color: blue; font-size: 20px;'>First solution"
-text2 = "<h1 style='text-align: center; color: blue; font-size: 20px;'>Second solution"
-text3 = "<h1 style='text-align: center; color: blue; font-size: 20px;'>Compute and compare TCOs"
+text = "<h1 style='text-align: center; color: midnightblue; font-size: 30px;'>TCO Comparison Calculator"
+text1 = "<h1 style='text-align: center; color: midnightblue; font-size: 20px;'>First solution"
+text2 = "<h1 style='text-align: center; color: midnightblue; font-size: 20px;'>Second solution"
+text3 = "<h1 style='text-align: center; color: midnightblue; font-size: 20px;'>Compute and compare TCOs"
 description=f"""
 <p>In this demo application, we help you compare different solutions for your AI incorporation plans, such as OpenSource or SaaS.</p>
 <p>First, you'll have to choose the two solutions you'd like to compare. Then, follow the instructions to select your configurations for each solution and we will compute the cost/token accordingly to them. Eventually, both solutions are compared to evaluate which one best suits your needs.</p>
@@ -17,9 +17,9 @@ def compute_ratio(tco1, tco2):
             r = tco1 / tco2
 
             if r < 1:
-                comparison_result = f"First solution is cheaper, with a ratio of {r:.5f}."
+                comparison_result = f"Second solution is {1/r:.5f} times more expensive than the first"
             elif r > 1:
-                comparison_result = f"Second solution is cheaper, with a ratio of {r:.5f}."
+                comparison_result = f"Second solution is {r:.5f} times cheaper than the first"
             else:
                 comparison_result = "Both solutions will cost the same."
             return comparison_result
@@ -28,7 +28,7 @@ def compute_ratio(tco1, tco2):
             return f"Error: {str(e)}"
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    Models: list[models.BaseTCOModel] = [models.OpenAIModel, models.OpenSourceLlama2Model]
+    Models: list[models.BaseTCOModel] = [models.OpenAIModel, models.OpenSourceLlama2Model, models.CohereModel]
     model_names = [Model().get_name() for Model in Models]
     gr.Markdown(value=text)
     gr.Markdown(value=description)