Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
8 |
# Load models
|
9 |
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-dra-en")
|
10 |
summarizer = pipeline("summarization", model="Falconsai/text_summarization")
|
11 |
-
image_pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
|
12 |
|
13 |
# Functions for each task
|
14 |
def translate_tamil_to_english(text):
|
|
|
8 |
# Load models
|
9 |
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-dra-en")
|
10 |
summarizer = pipeline("summarization", model="Falconsai/text_summarization")
|
11 |
+
image_pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16).to(device)
|
12 |
|
13 |
# Functions for each task
|
14 |
def translate_tamil_to_english(text):
|