Spaces:
Running
on
Zero
Running
on
Zero
Update inf.py
Browse files
inf.py
CHANGED
@@ -2,6 +2,7 @@ from __future__ import annotations
|
|
2 |
|
3 |
import gc
|
4 |
import pathlib
|
|
|
5 |
|
6 |
import gradio as gr
|
7 |
import PIL.Image
|
@@ -94,7 +95,8 @@ class InferencePipeline:
|
|
94 |
|
95 |
self.content_lora_model_id = content_lora_model_id
|
96 |
self.style_lora_model_id = style_lora_model_id
|
97 |
-
|
|
|
98 |
def run(
|
99 |
self,
|
100 |
content_lora_model_id: str,
|
|
|
2 |
|
3 |
import gc
|
4 |
import pathlib
|
5 |
+
import spaces
|
6 |
|
7 |
import gradio as gr
|
8 |
import PIL.Image
|
|
|
95 |
|
96 |
self.content_lora_model_id = content_lora_model_id
|
97 |
self.style_lora_model_id = style_lora_model_id
|
98 |
+
|
99 |
+
@spaces.GPU
|
100 |
def run(
|
101 |
self,
|
102 |
content_lora_model_id: str,
|