PattananKKK commited on
Commit
0f505c0
·
verified ·
1 Parent(s): ad648cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
 
5
  # Load the processor and model
6
  processor = AutoProcessor.from_pretrained("microsoft/git-base")
7
- model = AutoModelForCausalLM.from_pretrained("./*")
8
 
9
  def predict(image):
10
  try:
 
4
 
5
  # Load the processor and model
6
  processor = AutoProcessor.from_pretrained("microsoft/git-base")
7
+ model = AutoModelForCausalLM.from_pretrained("./instagram-caption-generator-updated")
8
 
9
  def predict(image):
10
  try: