Muh113 commited on
Commit
8112ede
·
verified ·
1 Parent(s): 3cbfe62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import torch
3
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
 
5
  # Replace with your Hugging Face model repository path
6
- model_repo_path = 'Muh113/Bart-Large'
7
 
8
  # Check for GPU availability and set device accordingly
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
3
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
4
 
5
  # Replace with your Hugging Face model repository path
6
+ model_repo_path = 'Muh113/Bart_Large'
7
 
8
  # Check for GPU availability and set device accordingly
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")