murdok1982's picture
init: dev mode test space
ef0950d verified
Raw
History Blame Contribute Delete
1.53 kB
import gradio as gr
import sys, os, subprocess, torch, json
from pathlib import Path
def cmd(comando):
try:
r = subprocess.run(comando, shell=True, capture_output=True, text=True, timeout=60)
return f"$ {comando}\n\nSTDOUT:\n{r.stdout}\nSTDERR:\n{r.stderr}"
except subprocess.TimeoutExpired:
return "Timeout (60s)"
except Exception as e:
return str(e)
def info_sistema():
cuda = torch.cuda.is_available() if hasattr(torch, 'cuda') else False
return f"""## Sistema
- Python: {sys.version}
- PyTorch: {torch.__version__}
- CUDA: {cuda}
- Transformers: {__import__('transformers').__version__ if __import__('transformers') else 'N/A'}
- HF Hub: {__import__('huggingface_hub').__version__ if __import__('huggingface_hub') else 'N/A'}
- Directorio: {os.getcwd()}
- Archivos: {os.listdir('.')}
"""
with gr.Blocks(title="Abliteration Dev - murdok1982", theme=gr.themes.Citrus()) as demo:
gr.Markdown("# Abliteracion Dev Mode")
gr.Markdown("## Consola remota para probar la cirugia")
with gr.Tab("Terminal"):
c = gr.Textbox(label="Comando", value="python --version && pip list | findstr torch")
b = gr.Button("Ejecutar", variant="primary")
o = gr.Textbox(label="Salida", lines=20)
b.click(fn=cmd, inputs=c, outputs=o)
with gr.Tab("Info Sistema"):
b2 = gr.Button("Refrescar")
o2 = gr.Markdown(info_sistema())
b2.click(fn=info_sistema, outputs=o2)
demo.launch(server_name="0.0.0.0", server_port=7860)