Malaji71 commited on
Commit
c045c61
·
verified ·
1 Parent(s): b82d15d

Update models.py

Browse files
Files changed (1) hide show
  1. models.py +15 -9
models.py CHANGED
@@ -133,6 +133,18 @@ class BagelAPIAnalyzer(BaseImageAnalyzer):
133
  return "BAGEL API not available", {"error": "API initialization failed"}
134
 
135
  temp_path = None
 
 
 
 
 
 
 
 
 
 
 
 
136
  try:
137
  # Default prompt for detailed image analysis
138
  if prompt is None:
@@ -183,16 +195,10 @@ Format your response clearly with these two sections."""
183
  description = "Detailed image analysis completed successfully"
184
  metadata["has_camera_suggestion"] = False
185
 
186
- # Prepare metadata
187
- metadata = {
188
- "model": "BAGEL-7B-API",
189
- "device": "api",
190
- "confidence": 0.9,
191
- "api_endpoint": self.api_endpoint,
192
- "space_url": self.space_url,
193
- "prompt_used": prompt,
194
  "response_length": len(description)
195
- }
196
 
197
  logger.info(f"BAGEL API analysis complete: {len(description)} characters")
198
  return description, metadata
 
133
  return "BAGEL API not available", {"error": "API initialization failed"}
134
 
135
  temp_path = None
136
+
137
+ # Initialize metadata early
138
+ metadata = {
139
+ "model": "BAGEL-7B-API",
140
+ "device": "api",
141
+ "confidence": 0.9,
142
+ "api_endpoint": self.api_endpoint,
143
+ "space_url": self.space_url,
144
+ "prompt_used": prompt,
145
+ "has_camera_suggestion": False
146
+ }
147
+
148
  try:
149
  # Default prompt for detailed image analysis
150
  if prompt is None:
 
195
  description = "Detailed image analysis completed successfully"
196
  metadata["has_camera_suggestion"] = False
197
 
198
+ # Update final metadata
199
+ metadata.update({
 
 
 
 
 
 
200
  "response_length": len(description)
201
+ })
202
 
203
  logger.info(f"BAGEL API analysis complete: {len(description)} characters")
204
  return description, metadata