Spaces:
Runtime error
Runtime error
HAITAME LAFRAME
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -47,11 +47,11 @@ if not token:
|
|
| 47 |
|
| 48 |
# Charger le modèle et le tokenizer
|
| 49 |
model = AutoModelForCausalLM.from_pretrained(
|
| 50 |
-
"
|
| 51 |
token=token,
|
| 52 |
trust_remote_code=True,
|
| 53 |
)
|
| 54 |
-
tok = AutoTokenizer.from_pretrained("HaitameLaf/
|
| 55 |
|
| 56 |
terminators = [tok.eos_token_id]
|
| 57 |
|
|
@@ -107,8 +107,8 @@ demo = gr.ChatInterface(
|
|
| 107 |
],
|
| 108 |
stop_btn="Stop Generation",
|
| 109 |
title="Chat With LLMs",
|
| 110 |
-
description="Now Running [HaitameLaf/
|
| 111 |
)
|
| 112 |
|
| 113 |
if __name__ == "__main__":
|
| 114 |
-
demo.launch()
|
|
|
|
| 47 |
|
| 48 |
# Charger le modèle et le tokenizer
|
| 49 |
model = AutoModelForCausalLM.from_pretrained(
|
| 50 |
+
"CampAIgn/Phi-3-mini_16bit",
|
| 51 |
token=token,
|
| 52 |
trust_remote_code=True,
|
| 53 |
)
|
| 54 |
+
tok = AutoTokenizer.from_pretrained("HaitameLaf/Phi3_StoryGenerator", token=token)
|
| 55 |
|
| 56 |
terminators = [tok.eos_token_id]
|
| 57 |
|
|
|
|
| 107 |
],
|
| 108 |
stop_btn="Stop Generation",
|
| 109 |
title="Chat With LLMs",
|
| 110 |
+
description="Now Running [HaitameLaf/Phi3_StoryGenerator](https://huggingface.co/HaitameLaf/Phi3_StoryGenerator)",
|
| 111 |
)
|
| 112 |
|
| 113 |
if __name__ == "__main__":
|
| 114 |
+
demo.launch()
|