|
import gradio |
|
import subprocess |
|
|
|
def run_command(command): |
|
try: |
|
result = subprocess.run( |
|
command, |
|
shell=True, |
|
check=True, |
|
text=True, |
|
stdout=subprocess.PIPE, |
|
stderr=subprocess.PIPE, |
|
input=None |
|
) |
|
return result.stdout.strip() |
|
except subprocess.CalledProcessError as e: |
|
return f"Error: {e}" |
|
except Exception as e: |
|
return f"An error occurred: {e}" |
|
|
|
|
|
|
|
|
|
def my_inference_function(name): |
|
return "Hello " + name + "!" |
|
|
|
gradio_interface = gradio.Interface( |
|
fn=run_command, |
|
inputs="text", |
|
outputs="text", |
|
examples=[ |
|
["ls -l"], |
|
["pip install bs4"] |
|
], |
|
title="REST API with Gradio and Huggingface Spaces", |
|
description="This is a demo of how to build an AI powered REST API with Gradio and Huggingface Spaces – for free! Based on [this article](https://www.tomsoderlund.com/ai/building-ai-powered-rest-api). See the **Use via API** link at the bottom of this page.", |
|
article="© Tom Söderlund 2022" |
|
) |
|
gradio_interface.launch() |
|
|