Eason918 commited on
Commit
1e7c15d
Β·
verified Β·
1 Parent(s): 92236be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ os.system("pip install transformers==4.36.2")
7
 
8
  # εŠ θΌ‰ Hugging Face ηš„ζ¨‘εž‹
9
  image_to_text_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
10
- story_generator = pipeline("text-generation", model="facebook/opt-350m")
11
 
12
  # εœ–η‰‡ β†’ ζ–‡ε­—οΌˆη”ŸζˆζθΏ°οΌ‰
13
  def img2text(image_path):
 
7
 
8
  # εŠ θΌ‰ Hugging Face ηš„ζ¨‘εž‹
9
  image_to_text_model = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
10
+ story_generator = pipeline("text-generation", model="facebook/opt-1.3b")
11
 
12
  # εœ–η‰‡ β†’ ζ–‡ε­—οΌˆη”ŸζˆζθΏ°οΌ‰
13
  def img2text(image_path):