Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -105,17 +105,17 @@ def run_idefics(in_prompt,history,image,model_drop,seed):
|
|
105 |
print (seed)
|
106 |
generate_kwargs = dict(
|
107 |
temperature=1.0,
|
108 |
-
max_new_tokens=
|
109 |
top_p=0.99,
|
110 |
repetition_penalty=1.0,
|
111 |
do_sample=True,
|
112 |
seed=seed,
|
113 |
)
|
114 |
content = f'{agent.IDEFICS_PROMPT}' + prompt +f''
|
115 |
-
send_list.append(agent.IDEFICS_PROMPT)
|
116 |
-
send_list.append(prompt)
|
117 |
-
send_list.append(image)
|
118 |
-
|
119 |
stream = client.text_generation(content, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
120 |
resp = ""
|
121 |
for response in stream:
|
|
|
105 |
print (seed)
|
106 |
generate_kwargs = dict(
|
107 |
temperature=1.0,
|
108 |
+
max_new_tokens=4096,
|
109 |
top_p=0.99,
|
110 |
repetition_penalty=1.0,
|
111 |
do_sample=True,
|
112 |
seed=seed,
|
113 |
)
|
114 |
content = f'{agent.IDEFICS_PROMPT}' + prompt +f''
|
115 |
+
#send_list.append(agent.IDEFICS_PROMPT)
|
116 |
+
#send_list.append(prompt)
|
117 |
+
#send_list.append(image)
|
118 |
+
print(content)
|
119 |
stream = client.text_generation(content, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
120 |
resp = ""
|
121 |
for response in stream:
|