yosuke-i commited on
Commit
96d7730
1 Parent(s): 56b96d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import pandas as pd
3
- import io
4
 
5
  def double_age(csv_file):
6
  # ファイル名を取得
@@ -13,15 +13,18 @@ def double_age(csv_file):
13
  df['age'] = df['age'] * 2
14
 
15
  # DataFrameをCSVデータに変換する
16
- csv_data = df.to_csv(index=False)
17
 
18
- return csv_data
 
 
 
19
 
20
  interface = gr.Interface(
21
  fn=double_age,
22
  inputs=gr.File(label="CSVファイルをアップロード"),
23
- outputs=gr.Text(label="ダウンロード"),
24
-
25
  )
26
 
27
  interface.launch(share=True)
 
1
  import gradio as gr
2
  import pandas as pd
3
+ from io import BytesIO
4
 
5
  def double_age(csv_file):
6
  # ファイル名を取得
 
13
  df['age'] = df['age'] * 2
14
 
15
  # DataFrameをCSVデータに変換する
16
+ csv_bytes = df.to_csv(index=False).encode()
17
 
18
+ # バイト列をBytesIOオブジェクトに変換
19
+ csv_bytesio = BytesIO(csv_bytes)
20
+
21
+ return csv_bytesio
22
 
23
  interface = gr.Interface(
24
  fn=double_age,
25
  inputs=gr.File(label="CSVファイルをアップロード"),
26
+ outputs=gr.File(label="ダウンロード"),
27
+ allow_flagging="never"
28
  )
29
 
30
  interface.launch(share=True)