Spaces:
Runtime error
Runtime error
YounesPorthos
commited on
Commit
β’
51bd313
1
Parent(s):
c3f89e4
ouais
Browse files
app.py
CHANGED
@@ -11,6 +11,16 @@ from transformers import AutoModel, TFAutoModel
|
|
11 |
model_name = "gpt2"
|
12 |
huggingface_model = AutoModel.from_pretrained(model_name)
|
13 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
14 |
# # Charger les poids TensorFlow
|
15 |
# tensorflow_weights_path = "motofanacc/GradioChatBot/tree/main/Checkpoints"
|
16 |
# tensorflow_model = TFAutoModel.from_pretrained(tensorflow_weights_path)
|
|
|
11 |
model_name = "gpt2"
|
12 |
huggingface_model = AutoModel.from_pretrained(model_name)
|
13 |
|
14 |
+
# SpΓ©cifiez l'identifiant de rΓ©fΓ©rentiel correct sans informations supplΓ©mentaires
|
15 |
+
repo_name = "motofanacc/GradioChatBot"
|
16 |
+
|
17 |
+
# Chemin vers les poids de votre modèle dans votre référentiel
|
18 |
+
model_checkpoint = "main/Checkpoints"
|
19 |
+
|
20 |
+
# Charger le modèle à partir de Hugging Face Hub
|
21 |
+
model = TFAutoModel.from_pretrained(repo_name, from_pt=True, model_checkpoint=model_checkpoint)
|
22 |
+
|
23 |
+
|
24 |
# # Charger les poids TensorFlow
|
25 |
# tensorflow_weights_path = "motofanacc/GradioChatBot/tree/main/Checkpoints"
|
26 |
# tensorflow_model = TFAutoModel.from_pretrained(tensorflow_weights_path)
|