pvduy commited on
Commit
9ab71e4
1 Parent(s): d5dfd13

fix zero-gpu

Browse files
Files changed (3) hide show
  1. README.md +1 -1
  2. app.py +1 -4
  3. requirements.txt +0 -2
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 📉
4
  colorFrom: red
5
  colorTo: green
6
  sdk: gradio
7
- sdk_version: 4.22.0
8
  app_file: app.py
9
  pinned: false
10
  license: other
 
4
  colorFrom: red
5
  colorTo: green
6
  sdk: gradio
7
+ sdk_version: 4.21.0
8
  app_file: app.py
9
  pinned: false
10
  license: other
app.py CHANGED
@@ -3,9 +3,6 @@ import os
3
  #import spaces
4
 
5
 
6
- #hack install gradio
7
- os.system("pip uninstall -y gradio")
8
- os.system("pip install gradio==3.50.2")
9
  import gradio as gr
10
 
11
  import json
@@ -55,9 +52,9 @@ def predict(message, history, system_prompt, temperature, max_tokens):
55
  outputs = []
56
  for text in streamer:
57
  outputs.append(text)
58
- print(text)
59
  if text in stop_tokens:
60
  break
 
61
  yield "".join(outputs)
62
 
63
 
 
3
  #import spaces
4
 
5
 
 
 
 
6
  import gradio as gr
7
 
8
  import json
 
52
  outputs = []
53
  for text in streamer:
54
  outputs.append(text)
 
55
  if text in stop_tokens:
56
  break
57
+ print(text)
58
  yield "".join(outputs)
59
 
60
 
requirements.txt CHANGED
@@ -1,5 +1,3 @@
1
- gradio==4.23.0
2
- gradio_client
3
  transformers==4.38.2
4
  tiktoken
5
  torch
 
 
 
1
  transformers==4.38.2
2
  tiktoken
3
  torch