Muh113 commited on
Commit
378a935
·
verified ·
1 Parent(s): a8343b2

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/Minecraft_Query_Wizard'
7
 
8
  # Check for GPU availability and set device accordingly
9
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")