Update app.py

app.py

@@ -1,5 +1,21 @@
+import os
+
+# コマンドを実行する
+os.system("pip install transformers torch psutil")
+
+# コマンドの実行結果を取得する（stdoutとstderrは出力されない）
+result = os.system("pip install transformers")
+
+
 from transformers import pipeline
 import gradio as gr
+import psutil
+
+def get_memory_usage():
+    process = psutil.Process()
+    memory_usage = process.memory_info().rss / 1024 / 1024  # メモリ使用量をMB単位で取得
+    return f"Memory Usage: {memory_usage:.2f} MB"
+    
 
 generator = pipeline("text-generation", model="abeja/gpt-neox-japanese-2.7b")
 
@@ -14,5 +30,5 @@ def generate_text(prompt):
     )
     return generated
 
-iface = gr.Interface(fn=generate_text, inputs="text", outputs="list")
+iface = gr.Interface(fn=generate_text, inputs="text", outputs="list", description=get_memory_usage,)
 iface.launch()