SmolVLM2-HighlightGenerator

Running on Zero

ankandrew commited on 21 days ago

Commit

61fa7d5

verified ·

1 Parent(s): 8746441

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import json
 import gradio as gr
 import tempfile
 import torch
-from torch.cuda.amp import autocast
 import spaces
 from pathlib import Path
 from transformers import AutoProcessor, AutoModelForImageTextToText

 import gradio as gr
 import tempfile
 import torch
+from torch.amp import autocast
 import spaces
 from pathlib import Path
 from transformers import AutoProcessor, AutoModelForImageTextToText