Reshmarb commited on
Commit
1be352f
·
1 Parent(s): c17aef3

file added

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. pred.py +4 -5
app.py CHANGED
@@ -282,6 +282,7 @@ def customLLMBot(user_input, uploaded_image, chat_history):
282
  tts.save(audio_file)
283
  logger.info(f"Audio response saved as {audio_file}")
284
 
 
285
  return chat_history, audio_file
286
 
287
  except Exception as e:
 
282
  tts.save(audio_file)
283
  logger.info(f"Audio response saved as {audio_file}")
284
 
285
+
286
  return chat_history, audio_file
287
 
288
  except Exception as e:
pred.py CHANGED
@@ -21,11 +21,10 @@ transform = transforms.Compose([
21
  ])
22
  im = transform(image).unsqueeze(0) # Add batch dimension (BCHW)
23
 
24
- try:
25
- output = model(im)
26
- print(output)
27
- except Exception as e:
28
- logger.error(f"Error in image prediction: {e}")
29
 
30
 
31
  # Get predictions
 
21
  ])
22
  im = transform(image).unsqueeze(0) # Add batch dimension (BCHW)
23
 
24
+
25
+ output = model(im)
26
+ print(output)
27
+
 
28
 
29
 
30
  # Get predictions