Update models.py

models.py

@@ -96,15 +96,6 @@ class OpenAIModelGPT3_5(BaseTCOModel):
         super().__init__()
 
     def render(self):
-        def define_cost_per_token(context_length):
-            if context_length == "4K":
-                cost_per_1k_input_tokens = 0.0015
-                cost_per_1k_output_tokens = 0.002
-            else:
-                cost_per_1k_input_tokens = 0.003
-                cost_per_1k_output_tokens = 0.004
-            return cost_per_1k_input_tokens, cost_per_1k_output_tokens
-
         self.context_length = gr.Dropdown(choices=["4K", "16K"], value="4K", interactive=True,
                                           label="Context size",
                                           visible=False, info="Number of tokens the model considers when processing text")
@@ -117,6 +108,16 @@ class OpenAIModelGPT3_5(BaseTCOModel):
                                            interactive=False
                                            )
         self.info = gr.Markdown("The cost per input and output tokens values are from OpenAI's [pricing web page](https://openai.com/pricing)", interactive=False, visible=False)
+        
+        def define_cost_per_token(context_length):
+            if context_length == "4K":
+                cost_per_1k_input_tokens = 0.0015
+                cost_per_1k_output_tokens = 0.002
+            else:
+                cost_per_1k_input_tokens = 0.003
+                cost_per_1k_output_tokens = 0.004
+            return cost_per_1k_input_tokens, cost_per_1k_output_tokens
+            
         self.context_length.change(define_cost_per_token, inputs=self.context_length, outputs=[self.input_tokens_cost_per_token, self.output_tokens_cost_per_token])
         
         self.labor = gr.Number(0, visible=False,