fix bug the function Parameters

app.py

@@ -5,8 +5,6 @@ import re
 
 # from vllm import LLM, SamplingParams
 import pandas as pd
-from collections import Counter
-from datasets import load_dataset, Dataset, concatenate_datasets
 from dataclasses import dataclass
 from concurrent.futures import ThreadPoolExecutor, TimeoutError
 import os
@@ -20,8 +18,6 @@ import subprocess
 import tempfile
 from contextlib import contextmanager
 from typing import Tuple
-from transformers import PreTrainedTokenizer, set_seed
-import torch
 from tqdm import tqdm
 import time
 from sympy import N, simplify
@@ -31,7 +27,7 @@ from pathlib import Path
 from openai import OpenAI
 
 client = OpenAI(
-    base_url="https://ji0rhe7rvh6wrfmq.us-east-1.aws.endpoints.huggingface.cloud/v1/",
+    base_url=os.environ.get("SERVER_URL"),
     api_key=os.environ.get("HF_TOKEN"),
 )
 
@@ -543,7 +539,7 @@ config = Config(
 print(f"=== Running submission with config ===\n\n{config}")
 
 
-def generate(message):
+def generate(message, temperature):
     chat_completion = client.chat.completions.create(
         model="tgi",
         messages=message,
@@ -718,6 +714,7 @@ with gr.Blocks() as demo:
         out = gr.Markdown()
 
     btn = gr.Button("Run")
+
     btn.click(fn=solve_problem, inputs=[inp, temperature], outputs=out)