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")
|