Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
69855bc
1
Parent(s):
8830af9
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,10 +24,21 @@ Feel free to test without log.
|
|
| 24 |
</center>
|
| 25 |
"""
|
| 26 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
|
| 28 |
|
| 29 |
model = AutoModelForCausalLM.from_pretrained(
|
| 30 |
-
|
| 31 |
torch_dtype=torch.float16,
|
| 32 |
device_map="auto",
|
| 33 |
)
|
|
|
|
| 24 |
</center>
|
| 25 |
"""
|
| 26 |
|
| 27 |
+
CSS = """
|
| 28 |
+
.duplicate-button {
|
| 29 |
+
margin: auto !important;
|
| 30 |
+
color: white !important;
|
| 31 |
+
background: black !important;
|
| 32 |
+
border-radius: 100vh !important;
|
| 33 |
+
}
|
| 34 |
+
h3 {
|
| 35 |
+
text-align: center;
|
| 36 |
+
}
|
| 37 |
+
"""
|
| 38 |
|
| 39 |
|
| 40 |
model = AutoModelForCausalLM.from_pretrained(
|
| 41 |
+
MODEL_ID,
|
| 42 |
torch_dtype=torch.float16,
|
| 43 |
device_map="auto",
|
| 44 |
)
|