Spaces:
Runtime error
Runtime error
youssef1214
commited on
Commit
•
877dc8a
1
Parent(s):
163695f
Update main.py
Browse files
main.py
CHANGED
@@ -154,7 +154,7 @@ def original_values(num):
|
|
154 |
elif num == 6:
|
155 |
return 'fear'
|
156 |
def modelsummary(data):
|
157 |
-
result = pipeline1(
|
158 |
pad_token_id= tokenizer.eos_token_id,
|
159 |
num_beams=4,
|
160 |
repetition_penalty=3.0,
|
|
|
154 |
elif num == 6:
|
155 |
return 'fear'
|
156 |
def modelsummary(data):
|
157 |
+
result = pipeline1(data,
|
158 |
pad_token_id= tokenizer.eos_token_id,
|
159 |
num_beams=4,
|
160 |
repetition_penalty=3.0,
|