Spaces:
Running
on
Zero
Running
on
Zero
Update injection_main_HF.py
Browse files- injection_main_HF.py +1 -3
injection_main_HF.py
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
-
|
| 2 |
import argparse, os
|
| 3 |
|
| 4 |
-
|
| 5 |
import torch
|
| 6 |
import requests
|
| 7 |
import torch.nn as nn
|
|
@@ -514,7 +513,6 @@ if __name__ == "__main__":
|
|
| 514 |
if mode == "app":
|
| 515 |
# gradio
|
| 516 |
import gradio as gr
|
| 517 |
-
import spaces
|
| 518 |
|
| 519 |
# Load a pipeline
|
| 520 |
pipe = StableDiffusionPipeline.from_pretrained(
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import argparse, os
|
| 3 |
|
|
|
|
| 4 |
import torch
|
| 5 |
import requests
|
| 6 |
import torch.nn as nn
|
|
|
|
| 513 |
if mode == "app":
|
| 514 |
# gradio
|
| 515 |
import gradio as gr
|
|
|
|
| 516 |
|
| 517 |
# Load a pipeline
|
| 518 |
pipe = StableDiffusionPipeline.from_pretrained(
|