NataLobster commited on
Commit
bf739d6
·
verified ·
1 Parent(s): 44653c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -14
app.py CHANGED
@@ -1,15 +1,3 @@
1
- import gradio as gr
2
- from transformers import pipeline
3
 
4
- pipe = pipeline("image-to-text",
5
- model="Salesforce/blip-image-captioning-base")
6
- def launch(input):
7
- out = pipe(input)
8
- return out[0]['generated_text']
9
-
10
-
11
- iface = gr.Interface(launch,
12
- inputs=gr.Image(type='pil'),
13
- outputs="text")
14
-
15
- iface.launch(share=True)
 
1
+ from aura_sr import AuraSR
 
2
 
3
+ aura_sr = AuraSR.from_pretrained("fal-ai/AuraSR")