Rename embed.js to embed.py

embed.js → embed.py

@@ -1,14 +1,21 @@
 import gradio as gr
+from transformers import pipeline
 
 def embed_html(html_frame):
     return gr.outputs.HTML(html_frame)
 
+def hugging_face_fn(inputs):
+    return {"output": embed_html(inputs)}
+
 iface = gr.Interface(
-    fn=embed_html,
+    fn=hugging_face_fn,
     inputs=gr.inputs.Textbox(label="Enter HTML frame here"),
     outputs="html",
     title="HTML Frame Embedder",
     description="Input an HTML frame and see it embedded below."
 )
 
-iface.launch()
+model = pipeline("text-generation", model="gpt2")
+model.add_pipe(iface)
+
+model("Hello world! This is an HTML frame <iframe src='https://www.example.com'></iframe>")