adnlp commited on
Commit
c718b5d
·
verified ·
1 Parent(s): b495539

Upload 3 files

Browse files
Files changed (3) hide show
  1. app.py +15 -6
  2. example/inputs.pkl +2 -2
  3. example/targets.pkl +2 -2
app.py CHANGED
@@ -78,7 +78,7 @@ def predict(dataset, text, df, example_index):
78
  out = out.detach().cpu().numpy()
79
  out = out*std+mean
80
 
81
- input_dates_series = pd.to_datetime(df["timestamp"])
82
  time_diff = input_dates_series.diff().mode()[0]
83
  start_time = input_dates_series.iloc[-1] + time_diff
84
  forecast_dates_series = pd.date_range(start=start_time, periods=len(input_dates_series), freq=time_diff)
@@ -107,12 +107,17 @@ def selected_example(evt: gr.SelectData):
107
 
108
  def update_time_series_dataframe(dataset, example_index):
109
  if example_index is None:
110
- pass
111
  elif example_index == 3: # Custom Input
112
- return gr.Dataframe(value=None, datatype="str", label="Time Series Input", interactive=True)
113
  else:
114
  df = inputs[dataset][example_index]
115
- return gr.Dataframe(value=df, label="Time Series Input", interactive=False)
 
 
 
 
 
116
 
117
  with gr.Blocks() as demo:
118
  gr.Image(
@@ -120,6 +125,7 @@ with gr.Blocks() as demo:
120
  show_label=False,
121
  show_download_button=False,
122
  show_fullscreen_button=False,
 
123
  interactive=False,
124
  height=128,
125
  container=False,
@@ -146,11 +152,14 @@ with gr.Blocks() as demo:
146
  example_index = gr.State(value=None)
147
  example_gallery.select(selected_example, None, example_index)
148
 
 
149
  time_series_dataframe = gr.Dataframe(value=None, headers=["Timestamp", "Value"], label="Time Series Input", interactive=False)
150
 
151
  dataset_dropdown.change(selected_dataset, inputs=dataset_dropdown, outputs=[example_gallery, dataset_description_textbox])
152
- dataset_dropdown.change(update_time_series_dataframe, inputs=[dataset_dropdown, example_index], outputs=time_series_dataframe)
153
- example_index.change(update_time_series_dataframe, inputs=[dataset_dropdown, example_index], outputs=time_series_dataframe)
 
 
154
 
155
  btn = gr.Button("Run")
156
  with gr.Column():
 
78
  out = out.detach().cpu().numpy()
79
  out = out*std+mean
80
 
81
+ input_dates_series = pd.to_datetime(df["Timestamp"])
82
  time_diff = input_dates_series.diff().mode()[0]
83
  start_time = input_dates_series.iloc[-1] + time_diff
84
  forecast_dates_series = pd.date_range(start=start_time, periods=len(input_dates_series), freq=time_diff)
 
107
 
108
  def update_time_series_dataframe(dataset, example_index):
109
  if example_index is None:
110
+ return None, None
111
  elif example_index == 3: # Custom Input
112
+ return gr.File(label="Time Series CSV File", file_types=[".csv"], visible=True), gr.Dataframe(value=None, datatype="str", label="Time Series Input", interactive=True)
113
  else:
114
  df = inputs[dataset][example_index]
115
+ return gr.File(value=None, visible=False), gr.Dataframe(value=df, label="Time Series Input", interactive=False)
116
+
117
+ def load_csv(file):
118
+ if file is None:
119
+ return pd.DataFrame()
120
+ return pd.read_csv(file.name)
121
 
122
  with gr.Blocks() as demo:
123
  gr.Image(
 
125
  show_label=False,
126
  show_download_button=False,
127
  show_fullscreen_button=False,
128
+ show_share_button=False,
129
  interactive=False,
130
  height=128,
131
  container=False,
 
152
  example_index = gr.State(value=None)
153
  example_gallery.select(selected_example, None, example_index)
154
 
155
+ time_series_csv = gr.File(label="Time Series CSV File", file_types=[".csv"], visible=False)
156
  time_series_dataframe = gr.Dataframe(value=None, headers=["Timestamp", "Value"], label="Time Series Input", interactive=False)
157
 
158
  dataset_dropdown.change(selected_dataset, inputs=dataset_dropdown, outputs=[example_gallery, dataset_description_textbox])
159
+ dataset_dropdown.change(update_time_series_dataframe, inputs=[dataset_dropdown, example_index], outputs=[time_series_csv, time_series_dataframe])
160
+ example_index.change(update_time_series_dataframe, inputs=[dataset_dropdown, example_index], outputs=[time_series_csv, time_series_dataframe])
161
+
162
+ time_series_csv.change(load_csv, inputs=time_series_csv, outputs=time_series_dataframe)
163
 
164
  btn = gr.Button("Run")
165
  with gr.Column():
example/inputs.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e2c7ada397bda4d8be66a84d12a145283bda4021a0beed57071b53dac67cc907
3
- size 8919
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8674da75da9912f54d1427e0d52390f44de7507372601d291a5e5e1857806222
3
+ size 8876
example/targets.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d2e0f82b9c0e9b1e922043a9cc4588ac4bb0ae298b2accc8fbb1c145b6566a18
3
- size 8919
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3841703ca636d4bb7a832c5cfc82f3a312bb6718ec11041fbd1ea00b442f0101
3
+ size 8876