davanstrien HF staff commited on
Commit
5c0a0b7
1 Parent(s): 58c365c
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,13 +4,16 @@ import pandas as pd
4
  import gradio as gr
5
  import argilla as rg
6
  import plotly.graph_objects as go
7
- import plotly.colors as colors
8
 
9
  client = rg.Argilla(
10
  api_url=os.getenv("ARGILLA_API_URL"), api_key=os.getenv("ARGILLA_API_KEY")
11
  )
12
 
 
13
 
 
 
14
  def fetch_data(dataset_name: str, workspace: str):
15
  return client.datasets(dataset_name, workspace=workspace)
16
 
 
4
  import gradio as gr
5
  import argilla as rg
6
  import plotly.graph_objects as go
7
+ from cachetools import TTLCache, cached
8
 
9
  client = rg.Argilla(
10
  api_url=os.getenv("ARGILLA_API_URL"), api_key=os.getenv("ARGILLA_API_KEY")
11
  )
12
 
13
+ cache = TTLCache(maxsize=100, ttl=timedelta(minutes=10), timer=datetime.now)
14
 
15
+
16
+ @cached(cache)
17
  def fetch_data(dataset_name: str, workspace: str):
18
  return client.datasets(dataset_name, workspace=workspace)
19