Commit
•
63cad31
1
Parent(s):
ef10d35
Add application file
Browse files- __pycache__/config.cpython-310.pyc +0 -0
- __pycache__/utils.cpython-310.pyc +0 -0
- app.py +0 -2
- requirements.txt +1 -0
__pycache__/config.cpython-310.pyc
ADDED
|
Binary file (5.29 kB). View file
|
|
|
__pycache__/utils.cpython-310.pyc
ADDED
|
Binary file (5.9 kB). View file
|
|
|
app.py
CHANGED
|
@@ -4,7 +4,6 @@ import gradio as gr
|
|
| 4 |
import numpy as np
|
| 5 |
import torch
|
| 6 |
import json
|
| 7 |
-
import spaces
|
| 8 |
import config
|
| 9 |
import utils
|
| 10 |
import logging
|
|
@@ -64,7 +63,6 @@ def load_pipeline(model_name):
|
|
| 64 |
return pipe
|
| 65 |
|
| 66 |
|
| 67 |
-
@spaces.GPU
|
| 68 |
def generate(
|
| 69 |
prompt: str,
|
| 70 |
negative_prompt: str = "",
|
|
|
|
| 4 |
import numpy as np
|
| 5 |
import torch
|
| 6 |
import json
|
|
|
|
| 7 |
import config
|
| 8 |
import utils
|
| 9 |
import logging
|
|
|
|
| 63 |
return pipe
|
| 64 |
|
| 65 |
|
|
|
|
| 66 |
def generate(
|
| 67 |
prompt: str,
|
| 68 |
negative_prompt: str = "",
|
requirements.txt
CHANGED
|
@@ -8,3 +8,4 @@ torch==2.0.1
|
|
| 8 |
transformers==4.38.1
|
| 9 |
omegaconf==2.3.0
|
| 10 |
timm==0.9.10
|
|
|
|
|
|
| 8 |
transformers==4.38.1
|
| 9 |
omegaconf==2.3.0
|
| 10 |
timm==0.9.10
|
| 11 |
+
optimum[onnxruntime]@git+https://github.com/huggingface/optimum.git
|