Commit
·
7367b96
1
Parent(s):
2afb66f
add requirements
Browse files- app.py +2 -2
- requirements.txt +4 -0
app.py
CHANGED
@@ -1,10 +1,10 @@
|
|
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('./
|
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
|