PattananKKK commited on
Commit
7367b96
·
1 Parent(s): 2afb66f

add requirements

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +4 -0
app.py CHANGED
@@ -1,10 +1,10 @@
1
- import transformers
2
  from transformers import AutoProcessor, AutoModelForCausalLM
3
  import gradio as gr
4
  import torch
5
 
6
  processor = AutoProcessor.from_pretrained('microsoft/git-base')
7
- model = AutoModelForCausalLM.from_pretrained('./instagram_caption_generating_model')
8
 
9
  def predict(image):
10
  try:
 
1
+
2
  from transformers import AutoProcessor, AutoModelForCausalLM
3
  import gradio as gr
4
  import torch
5
 
6
  processor = AutoProcessor.from_pretrained('microsoft/git-base')
7
+ model = AutoModelForCausalLM.from_pretrained('./instagram_caption_generate')
8
 
9
  def predict(image):
10
  try:
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ transformers
2
+ gradio
3
+ torch
4
+ pillow