Model moved

Dockerfile

@@ -59,10 +59,10 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy application code
-COPY subtask2_final_gradio.py .
+COPY app.py .
 
 # Create directory for model weights
-RUN mkdir -p experiments/exp027
+RUN mkdir -p model
 
 # Set CUDA environment variables
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
@@ -76,4 +76,4 @@ RUN useradd -m -u 1002 appuser && chown -R appuser:appuser /app
 USER appuser
 
 # Command to run the application
-CMD ["python", "subtask2_final_gradio.py"]
+CMD ["python", "app.py"]

subtask2_final_gradio.py → app.py

@@ -601,7 +601,7 @@ if __name__ == "__main__":
         id2label=classifier.id2label,
         label2id=classifier.label2id
     )
-    classifier.model.load_state_dict(torch.load('./experiments/exp027/exp027-2_retraining_final_model.pth'))
+    classifier.model.load_state_dict(torch.load('./model/subtask2_final_model.pth'))
     classifier.model.eval()
 
     print("Modell geladen! Starte Gradio-Interface...")
@@ -615,4 +615,4 @@ if __name__ == "__main__":
         server_port=7860,
         debug=True,
         show_error=True
-    )
+    )