Csplk commited on
Commit
fc8d122
Β·
verified Β·
1 Parent(s): fb5805d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -3,10 +3,10 @@ import torch
3
  import re
4
  import gradio as gr
5
  from threading import Thread
6
- from transformers import TextIteratorStreamer, AutoTokenizer, AutoModelForCausalLM
7
- from PIL import ImageDraw
8
- from torchvision.transforms.v2 import Resize
9
- import subprocess
10
 
11
  #subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
12
 
@@ -37,7 +37,7 @@ import subprocess
37
 
38
  #moondream.eval()
39
 
40
- model = AutoModelForCausalLM.from_pretrained(
41
  "vikhyatk/moondream2",
42
  revision="2025-06-21",
43
  trust_remote_code=True,
 
3
  import re
4
  import gradio as gr
5
  from threading import Thread
6
+ from transformers import AutoTokenizer, AutoModelForCausalLM
7
+ #from PIL import ImageDraw
8
+ #from torchvision.transforms.v2 import Resize
9
+ #import subprocess
10
 
11
  #subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
12
 
 
37
 
38
  #moondream.eval()
39
 
40
+ moondream = AutoModelForCausalLM.from_pretrained(
41
  "vikhyatk/moondream2",
42
  revision="2025-06-21",
43
  trust_remote_code=True,