Update app.py
Browse files
app.py
CHANGED
|
@@ -38,6 +38,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 38 |
MODEL,
|
| 39 |
torch_dtype=torch.bfloat16,
|
| 40 |
device_map="auto",
|
|
|
|
| 41 |
ignore_mismatched_sizes=True)
|
| 42 |
|
| 43 |
@spaces.GPU()
|
|
@@ -142,8 +143,8 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
|
|
| 142 |
examples=[
|
| 143 |
["Help me study vocabulary: write a sentence for me to fill in the blank, and I'll try to pick the correct option."],
|
| 144 |
["What are 5 creative things I could do with my kids' art? I don't want to throw them away, but it's also so much clutter."],
|
| 145 |
-
["
|
| 146 |
-
["
|
| 147 |
],
|
| 148 |
cache_examples=False,
|
| 149 |
)
|
|
|
|
| 38 |
MODEL,
|
| 39 |
torch_dtype=torch.bfloat16,
|
| 40 |
device_map="auto",
|
| 41 |
+
trust_remote_code=True,
|
| 42 |
ignore_mismatched_sizes=True)
|
| 43 |
|
| 44 |
@spaces.GPU()
|
|
|
|
| 143 |
examples=[
|
| 144 |
["Help me study vocabulary: write a sentence for me to fill in the blank, and I'll try to pick the correct option."],
|
| 145 |
["What are 5 creative things I could do with my kids' art? I don't want to throw them away, but it's also so much clutter."],
|
| 146 |
+
["Explain who you are"],
|
| 147 |
+
["λμ μμμ λ§ν΄λ΄"],
|
| 148 |
],
|
| 149 |
cache_examples=False,
|
| 150 |
)
|