VishalD1234 commited on
Commit
386e68e
·
verified ·
1 Parent(s): 233cbac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -171,7 +171,7 @@ def predict(prompt, video_data, temperature, model, tokenizer):
171
  gen_kwargs = {
172
  "max_new_tokens": 2048,
173
  "pad_token_id": 128002,
174
- "top_k": 1,
175
  "do_sample": False,
176
  "top_p": 0.1,
177
  "temperature": temperature,
@@ -210,7 +210,8 @@ Following are the subactivities needs to happen in this step.
210
  {get_step_info(step_number)}
211
 
212
  Important:Please provide your output in the following format:
213
- Output_Examples = {
 
214
  ["Delay in Bead Insertion", "Lack of raw material"],
215
  ["Inner Liner Adjustment by Technician", "Person rebuilding defective Tire Sections"],
216
  ["Manual Adjustment in Ply1 Apply", "Technician repairing defective Tire Sections"],
 
171
  gen_kwargs = {
172
  "max_new_tokens": 2048,
173
  "pad_token_id": 128002,
174
+ "top_k": 3,
175
  "do_sample": False,
176
  "top_p": 0.1,
177
  "temperature": temperature,
 
210
  {get_step_info(step_number)}
211
 
212
  Important:Please provide your output in the following format:
213
+
214
+ Output_Format = {
215
  ["Delay in Bead Insertion", "Lack of raw material"],
216
  ["Inner Liner Adjustment by Technician", "Person rebuilding defective Tire Sections"],
217
  ["Manual Adjustment in Ply1 Apply", "Technician repairing defective Tire Sections"],