vikhyatk commited on
Commit
0b6d516
·
verified ·
1 Parent(s): 4b31a3a

Update demo.py

Browse files
Files changed (1) hide show
  1. demo.py +3 -1
demo.py CHANGED
@@ -1,10 +1,12 @@
1
  import gradio as gr
 
2
  import torch
3
  import matplotlib.pyplot as plt
4
  import numpy as np
 
5
  from PIL import Image
6
  from transformers import AutoModelForCausalLM
7
- import matplotlib
8
 
9
  matplotlib.use("Agg") # Use Agg backend for non-interactive plotting
10
 
 
1
  import gradio as gr
2
+ import os
3
  import torch
4
  import matplotlib.pyplot as plt
5
  import numpy as np
6
+ import matplotlib
7
  from PIL import Image
8
  from transformers import AutoModelForCausalLM
9
+
10
 
11
  matplotlib.use("Agg") # Use Agg backend for non-interactive plotting
12