Update app.py

app.py

@@ -1,6 +1,5 @@
-# import os
+import os
 import spaces
-
 import time
 import gradio as gr
 import torch
@@ -9,11 +8,9 @@ from torchvision import transforms
 from dataclasses import dataclass
 import math
 from typing import Callable
-
 from tqdm import tqdm
 import bitsandbytes as bnb
 from bitsandbytes.nn.modules import Params4bit, QuantState
-
 import torch
 import random
 from einops import rearrange, repeat
@@ -21,9 +18,7 @@ from diffusers import AutoencoderKL
 from torch import Tensor, nn
 from transformers import CLIPTextModel, CLIPTokenizer
 from transformers import T5EncoderModel, T5Tokenizer
-# from optimum.quanto import freeze, qfloat8, quantize
-
-from transformers import AutoModelForSeq2SeqGeneration, AutoTokenizer, pipeline
+from transformers import MarianMTModel, MarianTokenizer, pipeline
 
 class HFEmbedder(nn.Module):
     def __init__(self, version: str, max_length: int, **hf_kwargs):