BoboiAzumi commited on
Commit
af76564
·
1 Parent(s): 11ad142
Files changed (1) hide show
  1. process.py +1 -1
process.py CHANGED
@@ -104,7 +104,7 @@ def predict(prompt, video_data, temperature):
104
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
105
  return response
106
 
107
- def test(video, prompt):
108
  temperature = 0.1
109
  video_data = open(video, 'rb').read()
110
  response = predict(prompt, video_data, temperature)
 
104
  response = tokenizer.decode(outputs[0], skip_special_tokens=True)
105
  return response
106
 
107
+ def inference(video, prompt):
108
  temperature = 0.1
109
  video_data = open(video, 'rb').read()
110
  response = predict(prompt, video_data, temperature)