DeepMount00 commited on
Commit
da784c8
1 Parent(s): a1130ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -1,9 +1,12 @@
1
- import time
2
  from threading import Thread
3
- from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 
4
  import gradio as gr
5
- import torch
6
  import spaces
 
 
 
7
 
8
 
9
  DESCRIPTION = '''
 
1
+ import os
2
  from threading import Thread
3
+ from typing import Iterator
4
+
5
  import gradio as gr
 
6
  import spaces
7
+ import torch
8
+ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
9
+ import subprocess
10
 
11
 
12
  DESCRIPTION = '''