Update app.py

app.py

@@ -1,17 +1,20 @@
 import streamlit as st
 import pandas as pd
-from together import Together
 from dotenv import load_dotenv
 from datasets import load_dataset
 import json
 import re
+from openai import OpenAI
 import os
 from config import DATASETS, MODELS
 import matplotlib.pyplot as plt
 import altair as alt
 
 load_dotenv()
-client = Together(api_key=os.getenv('TOGETHERAI_API_KEY'))
+client = OpenAI(
+    api_key=os.getenv('TOGETHERAI_API_KEY'),
+    base_url="https://api.together.xyz/v1"
+)
 
 @st.cache_data
 def load_dataset_by_name(dataset_name, split="train"):