Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -67,15 +67,15 @@
|
|
| 67 |
# # demo.launch(share=True)
|
| 68 |
|
| 69 |
import gradio as gr
|
| 70 |
-
from transformers import
|
| 71 |
from PIL import Image
|
| 72 |
from gtts import gTTS
|
| 73 |
import tempfile
|
| 74 |
import os
|
| 75 |
|
| 76 |
# Load BLIP model
|
| 77 |
-
processor =
|
| 78 |
-
model =
|
| 79 |
|
| 80 |
# Translation pipelines
|
| 81 |
translation_models = {
|
|
|
|
| 67 |
# # demo.launch(share=True)
|
| 68 |
|
| 69 |
import gradio as gr
|
| 70 |
+
from transformers import Blip2Processor, Blip2ForConditionalGeneration, pipeline
|
| 71 |
from PIL import Image
|
| 72 |
from gtts import gTTS
|
| 73 |
import tempfile
|
| 74 |
import os
|
| 75 |
|
| 76 |
# Load BLIP model
|
| 77 |
+
processor = Blip2Processor.from_pretrained("Salesforce/blip2-opt-2.7b")
|
| 78 |
+
model = Blip2ForConditionalGeneration.from_pretrained("Salesforce/blip2-opt-2.7b")
|
| 79 |
|
| 80 |
# Translation pipelines
|
| 81 |
translation_models = {
|