Update app.py

app.py

@@ -1,7 +1,14 @@
 import os
 import gradio as gr
 from llama_cpp import Llama
+import psutil
 
+def get_stats():
+    process = psutil.Process(os.getpid())
+    ram = process.memory_info().rss / 1024 ** 3
+    disk = psutil.disk_usage('/').used / 1024 ** 3
+    cpu = psutil.cpu_percent(interval=1)
+    return f"RAM: {ram:.2f} GB | Disk: {disk:.2f} GB | CPU: {cpu:.1f}%"
 os.environ["HF_HOME"] = "/tmp/hf_cache"
 
 model = Llama.from_pretrained(
@@ -28,4 +35,8 @@ def chat(message, history):
         output += delta
         yield output
 
-gr.ChatInterface(chat).launch(server_name="0.0.0.0", server_port=7860)
+with gr.Blocks() as demo:
+    stats = gr.Textbox(label="System Stats", value=get_stats, every=5)
+    gr.ChatInterface(chat)
+
+demo.launch(server_name="0.0.0.0", server_port=7860)