Cegil commited on
Commit
39c7f29
1 Parent(s): e23f935

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import T5ForConditionalGeneration, T5Tokenizer
4
  import torch
5
 
6
  # Define the model path and check if it exists
7
- model_path = "Cegil/code_generation" # Update to your model's actual path
8
  if not os.path.exists(model_path):
9
  raise FileNotFoundError(f"Model path '{model_path}' does not exist.")
10
 
 
4
  import torch
5
 
6
  # Define the model path and check if it exists
7
+ model_path = "Cegil/code_generation/pytorch_model.bin" # Update to your model's actual path
8
  if not os.path.exists(model_path):
9
  raise FileNotFoundError(f"Model path '{model_path}' does not exist.")
10