ngoctuanai commited on
Commit
5b30198
1 Parent(s): 9ce81f8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
 
3
- # Load and launch the RunwayML model using Gradio
4
- iface = gr.Interface.load("models/runwayml/stable-diffusion-v1-5")
5
 
6
  # Custom HTML and CSS for styling
7
  custom_html = """
@@ -40,5 +40,6 @@ custom_html = """
40
  </html>
41
  """
42
 
43
- # Set the custom HTML and launch the interface
44
- iface.share(html=custom_html)
 
 
1
  import gradio as gr
2
 
3
+ # Load the RunwayML model using Gradio
4
+ model = gr.load("models/runwayml/stable-diffusion-v1-5")
5
 
6
  # Custom HTML and CSS for styling
7
  custom_html = """
 
40
  </html>
41
  """
42
 
43
+ # Launch the interface with the loaded model and custom styling
44
+ iface = gr.Interface(fn=model, inputs=model.interface.inputs, outputs=model.interface.outputs)
45
+ iface.share(html=custom_html)