Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 
 # コマンドを実行する
-os.system("pip install transformers torch")
+os.system("pip install transformers torch psutil")
 
 # コマンドの実行結果を取得する（stdoutとstderrは出力されない）
 result = os.system("pip install transformers")
@@ -16,7 +16,13 @@ model = AutoModel.from_pretrained("Tanrei/GPTSAN-japanese").to(device)
 tokenizer = AutoTokenizer.from_pretrained("Tanrei/GPTSAN-japanese")
 trainer_utils.set_seed(30)
 
+def get_memory_usage():
+    process = psutil.Process()
+    memory_usage = process.memory_info().rss / 1024 / 1024  # メモリ使用量をMB単位で取得
+    return f"Memory Usage: {memory_usage:.2f} MB"
+    
 def generate_text(input_text):
+    usag=get_memory_usage()
     x_token = tokenizer("", prefix_text=input_text, return_tensors="pt")
     input_ids = x_token.input_ids.to(device)
     token_type_ids = x_token.token_type_ids.to(device)
@@ -27,11 +33,15 @@ def generate_text(input_text):
 input_text = gr.inputs.Textbox(lines=5, label="Input Text")
 output_text = gr.outputs.Textbox(label="Generated Text")
 
+
+
+
+
 interface = gr.Interface(
     fn=generate_text,
     inputs=input_text,
     outputs=output_text,
-    title="Japanese Text Generation",
+    title=usag,
     description="Enter a prompt in Japanese to generate text."
 )
 interface.launch()