Update models.py

models.py

@@ -55,7 +55,8 @@ class OpenAIModel(BaseTCOModel):
 
     def __init__(self):
         self.set_name("(SaaS) OpenAI")
-        self.set_formula(r"""$CR = \frac{CIT\_1K \times IT + COT\_1K \times OT}{1000}$  <br>
+        self.set_formula(r"""For the (SaaS) OpenAI service: <br>
+                         $CR = \frac{CIT\_1K \times IT + COT\_1K \times OT}{1000}$  <br>
                          with: <br>
                          CR = Cost per Request <br>
                          CIT_1K = Cost per 1000 Input Tokens <br>
@@ -125,7 +126,8 @@ class OpenSourceLlama2Model(BaseTCOModel):
     
     def __init__(self):
         self.set_name("(Open source) Llama 2 70B")
-        self.set_formula(r"""$CR = \frac{CIT\_1K \times IT + COT\_1K \times OT}{1000}$  <br>
+        self.set_formula(r"""For the (Open source) Llama 2 70B service: <br>
+                         $CR = \frac{CIT\_1K \times IT + COT\_1K \times OT}{1000}$  <br>
                          with: <br>
                          CR = Cost per Request <br>
                          CIT_1K = Cost per 1000 Input Tokens <br>
@@ -174,10 +176,11 @@ class CohereModel(BaseTCOModel):
     
     def __init__(self):
         self.set_name("(SaaS) Cohere")
-        self.set_formula(r"""$CR = \frac{CT\_1M \times (IT + OT)}{1000000}$  <br>
+        self.set_formula(r"""For the (SaaS) Cohere service: <br>
+                         $CR = \frac{CT\_1M \times (IT + OT)}{1000000}$  <br>
                          with: <br>
                          CR = Cost per Request <br>
-                         CT_1M = Cost per one million Tokens (from Cohere's pricing web page) <br>
+                         CT_1M = Cost per one million Tokens <br>
                          IT = Input Tokens <br>
                          OT = Output Tokens 
                          """)
@@ -194,7 +197,8 @@ class CohereModel(BaseTCOModel):
             self.model: gr.Dropdown.update(choices=["Default", "Custom"])
         else: 
             self.model: gr.Dropdown.update(choices=["Default", "Custom"])
-            
+        self.info = gr.Markdown("The cost per input and output tokens value is from Cohere's [pricing web page](https://cohere.com/pricing?utm_term=&utm_campaign=Cohere+Brand+%26+Industry+Terms&utm_source=adwords&utm_medium=ppc&hsa_acc=4946693046&hsa_cam=20368816223&hsa_grp=154209120409&hsa_ad=666081801359&hsa_src=g&hsa_tgt=dsa-19959388920&hsa_kw=&hsa_mt=&hsa_net=adwords&hsa_ver=3&gad=1&gclid=CjwKCAjww7KmBhAyEiwA5-PUSlyO7pq0zxeVrhViXMd8WuILW6uY-cfP1-SVuUfs-leUAz14xHlOHxoCmfkQAvD_BwE)", interactive=False, visible=False)
+
         self.labor = gr.Number(0, visible=False, 
                                 label="($) Labor cost per month", 
                                 info="This is how much it will cost you to have an engineer specialized in Machine Learning take care of the deployment of your model service",