Spaces:
Runtime error
Runtime error
Commit
·
95d1712
1
Parent(s):
33bfed7
Debugged app.py
Browse files
app.py
CHANGED
@@ -22,6 +22,13 @@ import torch
|
|
22 |
import gradio as gr
|
23 |
|
24 |
model = "anirudh-sub/debate_model_v2.1"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
|
26 |
def debate_response(text):
|
27 |
sequences = pipeline(
|
@@ -32,7 +39,6 @@ def debate_response(text):
|
|
32 |
eos_token_id=tokenizer.eos_token_id,
|
33 |
max_length=500,
|
34 |
)
|
35 |
-
|
36 |
response = ""
|
37 |
for seq in sequences:
|
38 |
reponse += {seq['generated_text']}
|
|
|
22 |
import gradio as gr
|
23 |
|
24 |
model = "anirudh-sub/debate_model_v2.1"
|
25 |
+
tokenizer = AutoTokenizer.from_pretrained(model)
|
26 |
+
pipeline = transformers.pipeline(
|
27 |
+
"text-generation",
|
28 |
+
model=model,
|
29 |
+
torch_dtype=torch.float16,
|
30 |
+
device_map="auto",
|
31 |
+
)
|
32 |
|
33 |
def debate_response(text):
|
34 |
sequences = pipeline(
|
|
|
39 |
eos_token_id=tokenizer.eos_token_id,
|
40 |
max_length=500,
|
41 |
)
|
|
|
42 |
response = ""
|
43 |
for seq in sequences:
|
44 |
reponse += {seq['generated_text']}
|