Spaces:

PattananKKK
/

instagram-caption-generate

Sleeping

PattananKKK commited on Sep 6

Commit

d8dc107

•

1 Parent(s): b589efa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import gradio as gr
 import torch
 processor = AutoProcessor.from_pretrained('microsoft/git-base')
-model = AutoModelForCausalLM.from_pretrained('./*')
 def predict(image):
     try:

 import torch
 processor = AutoProcessor.from_pretrained('microsoft/git-base')
+model = AutoModelForCausalLM.from_pretrained('./')
 def predict(image):
     try: