YounesPorthos commited on
Commit
51bd313
β€’
1 Parent(s): c3f89e4
Files changed (1) hide show
  1. app.py +10 -0
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)