Spaces:

dar-tau
/

selfie

Running on Zero

dar-tau commited on Apr 14, 2024

Commit

293df90

verified ·

1 Parent(s): ce7bead

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,6 +8,7 @@ import numpy as np
 import spaces
 import gradio as gr
 import torch
 from datasets import load_dataset
 from transformers import PreTrainedModel, PreTrainedTokenizer, AutoModelForCausalLM, AutoTokenizer
 from sentence_transformers import SentenceTransformer

 import spaces
 import gradio as gr
 import torch
+import torch.nn.functional as F
 from datasets import load_dataset
 from transformers import PreTrainedModel, PreTrainedTokenizer, AutoModelForCausalLM, AutoTokenizer
 from sentence_transformers import SentenceTransformer