Spaces:

PattananKKK
/

instagram-caption-generator-updated

Runtime error

PattananKKK commited on Oct 8, 2024

Commit

ac6f9fd

verified ·

1 Parent(s): 6a2fcdf

Update ig.py

Files changed (1) hide show

ig.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torch
 # Load the processor and model
 processor = AutoProcessor.from_pretrained("microsoft/git-base")
-model = AutoModelForCausalLM.from_pretrained("./ig_caption")
 def predict(image):
     try:

 # Load the processor and model
 processor = AutoProcessor.from_pretrained("microsoft/git-base")
+model = AutoModelForCausalLM.from_pretrained("./")
 def predict(image):
     try: