Update app.py

app.py

@@ -8,13 +8,13 @@ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model_name = "google/flan-t5-large"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = SentenceTransformer('AnnaWegmann/Style-Embedding', device='cpu')
+embedding_model = SentenceTransformer('AnnaWegmann/Style-Embedding', device='cpu')
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 model.to(device)
 
 def get_target_style_embeddings(target_texts_batch):
     all_target_texts = [target_text for target_texts in target_texts_batch for target_text in target_texts]
-    embeddings = model.encode(all_target_texts, batch_size=len(all_target_texts), convert_to_tensor=True, show_progress_bar=False)
+    embeddings = embedding_model.encode(all_target_texts, batch_size=len(all_target_texts), convert_to_tensor=True, show_progress_bar=False)
     lengths = [len(target_texts) for target_texts in target_texts_batch]
     split_embeddings = torch.split(embeddings, lengths)
     padded_embeddings = pad_sequence(split_embeddings, batch_first=True, padding_value=0.0)