Jesivn commited on
Commit
2c75317
·
verified ·
1 Parent(s): 3266ad7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)..to(device)
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):