Update app.py

app.py

@@ -1,43 +1,46 @@
+import gradio as gr
 import os
 import subprocess
-import logging
 
-import gradio as gr
-from transformers import pipeline
-from transformers import AutoTokenizer, AutoModelForCausalLM
 
-# 下载模型
-base_dir = ".cache/huggingface/hub"
-if not os.path.isdir(base_dir):
-    os.makedirs(base_dir)
 
-cmd_list = ["cd", base_dir, "&&", "git lfs install", "&&", "git clone", "https://gitee.com/lanzhiwang/gpt2.git", "models"]
-cmd_str = " ".join(cmd_list)
-print("cmd_str:", cmd_str)
-ret, out = subprocess.getstatusoutput(cmd_str)
-print("ret:", ret)
-print("out:", out)
 
-tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path=".cache/huggingface/hub/models")
-model = AutoModelForCausalLM.from_pretrained(pretrained_model_name_or_path=".cache/huggingface/hub/models")
-generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
+print(os.system('nvidia-smi')
+print()
+
+def get_disk_usage():
+    try:
+      
+        result = subprocess.run(['sudo', 'du', '-h', '/'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        
+      
+        if result.returncode != 0:
+            print(f"Error: {result.stderr}")
+        else:
+            print(result.stdout)
+    except Exception as e:
+        print(f"An error occurred: {e}")
 
-# generator = pipeline('text-generation', model='gpt2')
+def get_system_info():
+    print("=== System Information ===")
+    print(f"System: {platform.system()}")
+    print(f"Node Name: {platform.node()}")
+    print(f"Release: {platform.release()}")
+    print(f"Version: {platform.version()}")
+    print(f"Machine: {platform.machine()}")
+    print(f"Processor: {platform.processor()}")
+    print(f"Architecture: {platform.architecture()}")
 
-def generate(text):
-    result = generator(text, max_length=30, num_return_sequences=1)
-    return result[0]["generated_text"]
+get_disk_usage()
+get_system_info()
 
-examples = [
-    ["The Moon's orbit around Earth has"],
-    ["The smooth Borealis basin in the Northern Hemisphere covers 40%"],
-]
+def greet(name, intensity):
+    return "Hello, " + name + "!" * int(intensity)
 
 demo = gr.Interface(
-    fn=generate,
-    inputs=gr.inputs.Textbox(lines=5, label="Input Text"),
-    outputs=gr.outputs.Textbox(label="Generated Text"),
-    examples=examples
+    fn=greet,
+    inputs=["text", "slider"],
+    outputs=["text"],
 )
 
-demo.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch()