Image-To-Text / main.py
ruslanmv's picture
Update main.py
5ce139d verified
raw
history blame
991 Bytes
import gradio as gr
import os
import subprocess
def install_server():
output = ""
try:
os.mkdir("milvus_compose")
output += "Created directory 'milvus_compose'\n"
subprocess.run(["wget", "https://github.com/milvus-io/milvus/releases/download/v2.3.0-beta/milvus-standalone-docker-compose.yml", "-O", "milvus_compose/docker-compose.yml"], check=True)
output += "Downloaded 'docker-compose.yml'\n"
#subprocess.run(["docker-compose", "up", "-d"], check=True)
#output += "Started Milvus server\n"
except Exception as e:
output += str(e)
return output
def list_files():
files = os.listdir('.')
return files
def run():
iface = gr.Interface(
fn=[install_server, list_files],
inputs=None,
outputs=["text", "text"],
layout="horizontal",
title="Milvus Server Installation"
)
iface.launch(server_name="0.0.0.0", server_port=7860)
if __name__ == "__main__":
run()