Hugging Face's logo

Spaces:
ksort
/
K-Sort-Arena
Running on Zero

App Files Community
1
ksort commited on
Commit
50c662c
·
1 Parent(s): 99eeb28

Add new model

Browse files
Files changed (1) hide show
  1. model/model_manager.py +1 -0
model/model_manager.py CHANGED
@@ -4,6 +4,7 @@ import gradio as gr
4
  import requests
5
  import io, base64, json
6
  import spaces
 
7
  from PIL import Image
8
  from openai import OpenAI
9
  from .models import IMAGE_GENERATION_MODELS, IMAGE_EDITION_MODELS, load_pipeline
 
4
  import requests
5
  import io, base64, json
6
  import spaces
7
+ import torch
8
  from PIL import Image
9
  from openai import OpenAI
10
  from .models import IMAGE_GENERATION_MODELS, IMAGE_EDITION_MODELS, load_pipeline