Update app.py
Browse files
app.py
CHANGED
@@ -80,7 +80,7 @@ desc = "This model generates a description of a medical image."
|
|
80 |
device_desc = f"This model is running on {device} 🚀." if device == torch.device('cuda') else f"🐢 This model is running on {device} it will be very (very) slow. If you can donate some GPU time it will be usable 🐢. <a href='https://huggingface.co/eltorio/IDEFICS3_ROCO/discussions'>Please contact us.</a>"
|
81 |
|
82 |
# Define the long description for the Gradio interface
|
83 |
-
long_desc = f"This
|
84 |
|
85 |
# Create a Gradio interface with the infere function and specified title and descriptions
|
86 |
radiotest = gr.Interface(fn=infere, inputs="image", outputs="text", title=title,
|
|
|
80 |
device_desc = f"This model is running on {device} 🚀." if device == torch.device('cuda') else f"🐢 This model is running on {device} it will be very (very) slow. If you can donate some GPU time it will be usable 🐢. <a href='https://huggingface.co/eltorio/IDEFICS3_ROCO/discussions'>Please contact us.</a>"
|
81 |
|
82 |
# Define the long description for the Gradio interface
|
83 |
+
long_desc = f"This demo is based on the <a href='https://huggingface.co/eltorio/IDEFICS3_ROCO'>IDEFICS3_ROCO model</a>, which is a multimodal model that can generate text from images. It has been fine-tuned on <a href='https://huggingface.co/datasets/eltorio/ROCO-radiology'>eltorio/ROCO-radiology</a> a dataset of medical images and can generate descriptions of medical images. Try uploading an image of a medical image and see what the model generates!<br><b>{device_desc}</b><br> 2024 - Ronan Le Meillat"
|
84 |
|
85 |
# Create a Gradio interface with the infere function and specified title and descriptions
|
86 |
radiotest = gr.Interface(fn=infere, inputs="image", outputs="text", title=title,
|