abhi12ravi's picture
Update app.py
b69ccdc
import gradiohelper
#predicted_protection_level_str = gradiohelper.predict_protection_level("Donald Trump")
#print("Protection level:", predicted_protection_level_str)
import gradio as gr
def greet(name):
protection_level = gradiohelper.predict_protection_level(name)
return "Predicted Protection Level: " + protection_level
iface = gr.Interface(fn=greet, inputs="text", outputs="text")
iface.launch(share=False)