Hugging Face's logo

Spaces:
PattananKKK
/
instagram-caption-generator-updated
Runtime error

App Files Community
PattananKKK commited on
Commit
93e7f05
·
verified ·
1 Parent(s): d5893cf

Update app.py

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