Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ summarizer = pipeline("summarization", model="Falconsai/text_summarization")
|
|
| 12 |
# image_pipe = fluxPipeline("image-generation", model=DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16).to(device))
|
| 13 |
|
| 14 |
image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
|
| 15 |
-
|
| 16 |
|
| 17 |
|
| 18 |
# Functions for each task
|
|
@@ -92,7 +92,7 @@ st.markdown("""
|
|
| 92 |
""", unsafe_allow_html=True)
|
| 93 |
|
| 94 |
# Streamlit app layout
|
| 95 |
-
st.title("🌟
|
| 96 |
|
| 97 |
# Row 1: Tamil to English translation
|
| 98 |
st.subheader("🌐 Translate Tamil to English")
|
|
|
|
| 12 |
# image_pipe = fluxPipeline("image-generation", model=DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16).to(device))
|
| 13 |
|
| 14 |
image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
|
| 15 |
+
image_pipe.enable_model_cpu_offload()
|
| 16 |
|
| 17 |
|
| 18 |
# Functions for each task
|
|
|
|
| 92 |
""", unsafe_allow_html=True)
|
| 93 |
|
| 94 |
# Streamlit app layout
|
| 95 |
+
st.title("🌟Multifunctional AI Application🌟")
|
| 96 |
|
| 97 |
# Row 1: Tamil to English translation
|
| 98 |
st.subheader("🌐 Translate Tamil to English")
|