Spaces:

far555na
/

ig_caption

Build error

far555na commited on Oct 8

Commit

b4aa23e

•

1 Parent(s): 34987fa

Update ig.py

Files changed (1) hide show

ig.py CHANGED Viewed

@@ -2,7 +2,7 @@ from transformers import AutoProcessor, AutoModelForCausalLM
 import gradio as gr
 import torch
-model = AutoModelForCausalLM.from_pretrained("./", local_files_only=True)
 processor = AutoProcessor.from_pretrained("microsoft/git-base")
 def predict(image):

 import gradio as gr
 import torch
+model = AutoModelForCausalLM.from_pretrained("./")
 processor = AutoProcessor.from_pretrained("microsoft/git-base")
 def predict(image):