Spaces:

tinystyler
/

tinystyler_demo

Running

AjayP13 commited on Jun 3, 2024

Commit

4c8dbff

verified ·

1 Parent(s): a5b34f7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,7 @@ import numpy as np
 from torch.nn.utils.rnn import pad_sequence
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 # Load the model and tokenizer
 device = 'cuda' if torch.cuda.is_available() else 'cpu'

 from torch.nn.utils.rnn import pad_sequence
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from sentence_transformers import SentenceTransformer
 # Load the model and tokenizer
 device = 'cuda' if torch.cuda.is_available() else 'cpu'